Update app.py

app.py

@@ -4,12 +4,17 @@ import torch
 
 # Load the quantized model from Hugging Face
 model_id = "Intel/sd-1.5-square-quantized"
-pipe = OVStableDiffusionPipeline.from_pretrained(model_id, export=False, library_name="diffusers")
+pipe = OVStableDiffusionPipeline.from_pretrained(
+    model_id,
+    export=False,
+    trust_remote_code=True,  # Allow custom code if needed
+    ov_config={"CACHE_DIR": ""}  # Optional: disable caching if it’s causing issues
+)
 
 # Define the inference function
 def generate_image(prompt):
     # Generate the image
-    image = pipe(prompt, num_inference_steps=5, guidance_scale=7.5).images[0]
+    image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5).images[0]
     return image
 
 # Create the Gradio interface