Update app.py

app.py

@@ -3,23 +3,28 @@ import gradio as gr
 from diffusers import StableDiffusionPipeline
 import torch
 
-# Load model once
-model_id = "runwayml/stable-diffusion-v1-5"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
-pipe = pipe.to("cpu")  # force CPU to run free on Spaces
+# Load Stable Diffusion pipeline (open source)
+pipe = StableDiffusionPipeline.from_pretrained(
+    "runwayml/stable-diffusion-v1-5", 
+    torch_dtype=torch.float32
+)
+
+# Force CPU (free Spaces usually don’t allow GPU)
+pipe.to("cpu")
 
-# Simple function for text-to-image
-def generate_image(prompt: str):
-    image = pipe(prompt, num_inference_steps=20, guidance_scale=7.5).images[0]
+def generate_image(prompt):
+    image = pipe(prompt, guidance_scale=7.5).images[0]
     return image
 
-# API-only mode, no full UI
+# Gradio UI for testing
 demo = gr.Interface(
     fn=generate_image,
     inputs=gr.Textbox(label="Enter your prompt"),
-    outputs=gr.Image(type="pil"),
-    live=False,
+    outputs=gr.Image(label="Generated Image"),
+    title="Stable Diffusion Text-to-Image",
+    description="Enter a prompt and generate an image using Stable Diffusion."
 )
 
-# Make it lightweight & responsive (minimal layout)
-demo.launch(server_name="0.0.0.0", server_port=7860, show_api=True)
+if __name__ == "__main__":
+    demo.launch()
+