Update app.py

app.py

@@ -2,9 +2,10 @@ import gradio as gr
 from diffusers import StableDiffusionPipeline
 import torch
 
+# Use a compatible model ID
 model_id = "CompVis/stable-diffusion-v1-4"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)  # Use float32 if float16 causes issues
+pipe = pipe.to("cuda")  # Ensure your environment supports CUDA; if not, remove this line
 
 def generate_image(prompt):
     image = pipe(prompt).images[0]
@@ -12,12 +13,10 @@ def generate_image(prompt):
 
 iface = gr.Interface(
     fn=generate_image,
-    inputs=gr.Textbox(label="prompt", placeholder="Enter your prompt here..."),
+    inputs=gr.Textbox(label="Prompt", placeholder="Enter your prompt here..."),
     outputs=gr.Image(type="pil", label="Generated Image"),
-    title="Speech2Image Generator (Fast API Powered)",
+    title="Speech2Image Generator (FastAPI Powered)",
     theme="default",
 )
 
 iface.launch()
-
-