Update app.py

app.py

@@ -3,8 +3,7 @@ from diffusers import StableDiffusionPipeline
 import gradio as gr
 
 # Load model (optimized for Hugging Face Spaces)
-model_id = "runwayml/stable-diffusion-v1-5"
-
+model_id = "stabilityai/sd-turbo"
 pipe = StableDiffusionPipeline.from_pretrained(
     model_id,
     torch_dtype=torch.float16
@@ -19,12 +18,13 @@ pipe.enable_attention_slicing()
 
 # Image generation function
 def generate_image(prompt):
-    if prompt.strip() == "":
-        return "Please enter a prompt"
-
-    image = pipe(prompt).images[0]
+    image = pipe(
+        prompt,
+        num_inference_steps=20,   # reduce from default (~50)
+        guidance_scale=7.5
+    ).images[0]
     return image
-
+    
 # Gradio UI
 interface = gr.Interface(
     fn=generate_image,