Update app.py

app.py

@@ -1,10 +1,17 @@
 import torch
 from diffusers import StableDiffusionPipeline
+import gradio as gr
 
-pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16
-)
-pipe = pipe.to("cpu")
+# Load model on CPU (use torch_dtype=torch.float32 for CPU compatibility)
+pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float32).to("cpu")
 
-prompt = "a photo of an astronaut riding a horse on mars"
-image = pipe(prompt).images[0]
-image
+# Function to generate image
+def generate_image(prompt):
+    image = pipe(prompt).images[0]
+    return image
+
+# Gradio interface
+iface = gr.Interface(fn=generate_image, inputs="text", outputs="image", title="Stable Diffusion Image Generator")
+
+# Launch app
+iface.launch()