Update app.py

app.py

@@ -2,10 +2,10 @@ import gradio as gr
 from diffusers import StableDiffusionPipeline
 import torch
 
-# Model Load
+# Model Load (Free Space safe version)
 model_id = "runwayml/stable-diffusion-v1-5"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe = pipe.to("cuda")  # GPU use
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
+pipe = pipe.to("cpu")  # Free Space में GPU नहीं होने पर CPU use
 
 def generate_image(prompt):
     image = pipe(prompt).images[0]
@@ -14,10 +14,14 @@ def generate_image(prompt):
 # Gradio Interface
 iface = gr.Interface(
     fn=generate_image,
-    inputs=gr.Textbox(label="Type your image description"),
+    inputs=gr.Textbox(lines=2, placeholder="Type your image description here...", label="Prompt"),
     outputs=gr.Image(label="Generated Image"),
     title="AI Image Generator",
-    description="Type anything and get AI-generated image instantly!"
+    description="""
+Type any text and get beautiful AI-generated images instantly! 
+Free low-res images available. Perfect for posters, social media, and creative art.
+""",
+    allow_flagging="never"
 )
 
 iface.launch()