Update app.py

app.py

@@ -1,14 +1,20 @@
 import gradio as gr
-from optimum.intel.openvino import OVStableDiffusionPipeline
+from diffusers import StableDiffusionPipeline
 import torch
 
-# Load the quantized model from Hugging Face
+# Load the model
 model_id = "Intel/sd-1.5-square-quantized"
-pipe = OVStableDiffusionPipeline.from_pretrained(model_id, export=False, library_name="diffusers")
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+
+# Optional: Move to CPU (default in Spaces) or enable OpenVINO if available
+try:
+    from optimum.intel import OVStableDiffusionPipeline
+    pipe = OVStableDiffusionPipeline.from_pretrained(model_id, export=False, library_name="diffusers")
+except (ImportError, ModuleNotFoundError):
+    print("Falling back to standard diffusers pipeline (no OpenVINO optimization).")
 
 # Define the inference function
 def generate_image(prompt):
-    # Generate the image
     image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5).images[0]
     return image