Update app.py

app.py

@@ -2,24 +2,34 @@ import gradio as gr
 from diffusers import StableDiffusionPipeline
 import torch
 
-# Model Load (Free Space safe version)
+# Model Load (Free Space friendly)
 model_id = "runwayml/stable-diffusion-v1-5"
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
-pipe = pipe.to("cpu")  # Free Space में GPU नहीं होने पर CPU use
+pipe = pipe.to("cpu")  # Free Space CPU safe
 
-def generate_image(prompt):
-    image = pipe(prompt).images[0]
+# Function for fast low-res preview + high-res paid option
+def generate_image(prompt, high_res=False):
+    if high_res:
+        # Paid high-res: slower
+        image = pipe(prompt, height=512, width=512).images[0]
+    else:
+        # Free low-res: faster
+        image = pipe(prompt, height=256, width=256).images[0]
     return image
 
 # Gradio Interface
 iface = gr.Interface(
     fn=generate_image,
-    inputs=gr.Textbox(lines=2, placeholder="Type your image description here...", label="Prompt"),
+    inputs=[
+        gr.Textbox(lines=2, placeholder="Type your image description here...", label="Prompt"),
+        gr.Checkbox(label="High-Resolution (Paid Feature)")
+    ],
     outputs=gr.Image(label="Generated Image"),
-    title="AI Image Generator",
+    title="AI Image Generator (Fast Preview)",
     description="""
-Type any text and get beautiful AI-generated images instantly! 
-Free low-res images available. Perfect for posters, social media, and creative art.
+Type any text to get beautiful AI-generated images instantly!
+Free low-res images for fast preview.
+Check High-Resolution for paid download.
 """,
     allow_flagging="never"
 )