Update app.py

app.py

@@ -1,13 +1,19 @@
-from diffusers import DiffusionPipeline, LCMScheduler, Transformer2DModelOutput
+from diffusers import DiffusionPipeline, LCMScheduler
+from diffusers.models.modeling_outputs import Transformer2DModelOutput
 from optimum.intel import OVStableDiffusionXLPipeline
 import gradio as gr
 
-
-
+# Loading the model
 model_id = "stabilityai/stable-diffusion-xl-base-1.0"
 pipe = DiffusionPipeline.from_pretrained(model_id)
+
+# Setting the scheduler
 pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)  
+
+# Loading LoRA weights
 pipe.load_lora_weights("latent-consistency/lcm-lora-sdxl")
+
+# Converting the model to OpenVINO format
 pipeline = OVStableDiffusionXLPipeline.from_pretrained(
     pipe,  
     export=True