Update app.py

app.py

@@ -1,36 +1,25 @@
 
-import torch
-from diffusers import StableDiffusionPipeline
 import gradio as gr
+from diffusers import StableDiffusionPipeline
+import torch
 
-# Load a lighter model for CPU
-model_id = "stabilityai/stable-diffusion-2-1-base"
-
-pipe = StableDiffusionPipeline.from_pretrained(
-    model_id,
-    torch_dtype=torch.float32
-).to("cpu")
-
-# CPU-friendly optimization
-pipe.enable_attention_slicing()
+# Load model once
+model_id = "runwayml/stable-diffusion-v1-5"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
+pipe = pipe.to("cpu")  # force CPU to run free on Spaces
 
-def generate(prompt):
-    image = pipe(
-        prompt,
-        num_inference_steps=12,   # keep low for speed
-        guidance_scale=7,        # balanced CFG
-        width=384,               # smaller resolution
-        height=384
-    ).images[0]
+# Simple function for text-to-image
+def generate_image(prompt: str):
+    image = pipe(prompt, num_inference_steps=20, guidance_scale=7.5).images[0]
     return image
 
-with gr.Blocks(css=".gradio-container {max-width: 800px; margin: auto;}") as demo:
-    gr.Markdown("## 🎨 AI Image Generator (CPU)\nType your prompt and generate in ~20s.")
-
-    prompt = gr.Textbox(label="Prompt", placeholder="A futuristic city at night")
-    output = gr.Image(type="pil", label="Generated Image")
-    btn = gr.Button("🚀 Generate")
-
-    btn.click(fn=generate, inputs=prompt, outputs=output)
+# API-only mode, no full UI
+demo = gr.Interface(
+    fn=generate_image,
+    inputs=gr.Textbox(label="Enter your prompt"),
+    outputs=gr.Image(type="pil"),
+    live=False,
+)
 
-demo.launch()
+# Make it lightweight & responsive (minimal layout)
+demo.launch(server_name="0.0.0.0", server_port=7860, show_api=True)