Update app.py

app.py

@@ -2,26 +2,33 @@ import gradio as gr
 from diffusers import StableDiffusionPipeline
 import torch
 
-# ✅ Correct model ID
+# Load the model
 model_id = "runwayml/stable-diffusion-v1-5"
 
-# ✅ Removed the bad 'revision' argument
+# ✅ Detect whether GPU is available
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+# ✅ Load pipeline without 'use_auth_token'
 pipe = StableDiffusionPipeline.from_pretrained(
     model_id,
-    torch_dtype=torch.float16,
-    use_auth_token=True  # Optional on Hugging Face Spaces
-).to("cuda")
+    torch_dtype=torch.float16 if device == "cuda" else torch.float32
+)
+
+# ✅ Move model to correct device
+pipe = pipe.to(device)
 
+# Inference function
 def generate_image(prompt):
     image = pipe(prompt).images[0]
     return image
 
+# UI
 demo = gr.Interface(
     fn=generate_image,
-    inputs=gr.Textbox(label="Enter image description", placeholder="e.g. A fantasy tree in the fog"),
+    inputs=gr.Textbox(label="Enter image description", placeholder="e.g. A magical tree in a glowing forest"),
     outputs=gr.Image(type="pil"),
     title="Text to Image Generator",
-    description="Enter a description to generate an image using Stable Diffusion"
+    description="Enter a prompt and generate an image using Stable Diffusion"
 )
 
 if __name__ == "__main__":