Update app.py

app.py

@@ -1,23 +1,22 @@
 from diffusers import DiffusionPipeline, LCMScheduler
+from optimum.intel import OVStableDiffusionXLPipeline
 import gradio as gr
-import torch
 
 
-pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", 
-                                         torch_dtype=torch.float16)
+model_id = "stabilityai/stable-diffusion-xl-base-1.0"
+pipe = DiffusionPipeline.from_pretrained(model_id)
 pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
 pipe.load_lora_weights("latent-consistency/lcm-lora-sdxl")
-pipe = pipe.to("cpu")
 
+pipeline = OVStableDiffusionXLPipeline.from_pretrained(
+    pipe,
+    export=True 
+)
 
 def generate_images(prompt, batch_size):
     images = []
     for _ in range(batch_size):
-        results = pipe(
-            prompt=prompt,
-            num_inference_steps=4,
-            guidance_scale=0.0,
-        )
+        results = pipeline(prompt)
         images.append(results.images[0])
     return images
 
@@ -28,7 +27,7 @@ iface = gr.Interface(
         gr.Slider(label="Number of Images", minimum=1, maximum=12, step=1, value=1)
     ],
     outputs=gr.Gallery(label="Generated Images"),
-    title="SuperFast SDXL Image Generation"
+    title="SuperFast SDXL Generation on CPU"
 )
 
-iface.launch()
+iface.launch(debug=True)