Update app.py

app.py

@@ -10,7 +10,7 @@ pipe = DiffusionPipeline.from_pretrained(model_id)
 pipe = pipe.to("cpu")  # Force CPU usage
 
 # 3. FASTER GENERATION SETTINGS
-def generate_image(prompt, negative_prompt="", steps=15):
+def generate_image(prompt, negative_prompt="", steps=13):
     return pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
@@ -20,13 +20,13 @@ def generate_image(prompt, negative_prompt="", steps=15):
 
 # 4. STREAMLINED UI
 with gr.Blocks() as demo:
-    gr.Markdown("# ⚡ Lightning-Fast AI Image Generator")
+    gr.Markdown("# Fast AI Image Generator using OFA Small")
     
     with gr.Row():
         with gr.Column():
-            prompt = gr.Textbox(label="Your Prompt", value="a cat astronaut")
-            negative = gr.Textbox(label="Avoid (Optional)", value="blurry, deformed")
-            steps = gr.Slider(1, 30, value=10, label="Quality Steps")
+            prompt = gr.Textbox(label="Your Prompt", value="a beautiful flower")
+            negative = gr.Textbox(label="Avoid (Optional)", value="low-resolution")
+            steps = gr.Slider(1, 30, value=13, label="Quality Steps")
             btn = gr.Button("Generate →")
         
         output = gr.Image(label="Result", height=400)
@@ -35,8 +35,8 @@ with gr.Blocks() as demo:
     
     gr.Examples(
         examples=[
-            ["cyberpunk cityscape at night, neon lights", "people, text", 12],
-            ["watercolor painting of a forest", "photorealistic, humans", 8]
+            ["cityscape at night, red lights", "people", 12],
+            ["watercolor painting of a flower", "photorealistic", 8]
         ],
         inputs=[prompt, negative, steps]
     )