Upload app.py with huggingface_hub

app.py

@@ -6,7 +6,6 @@ import spaces
 import torch
 import gradio as gr
 import random
-import time
 
 from mage_flow import MageFlowPipeline
 from mage_flow.models.modules._attn_backend import set_attn_backend
@@ -22,10 +21,12 @@ set_attn_backend("sdpa")
 
 
 def _estimate_duration(prompt, negative_prompt, height, width, steps, cfg, seed, *args, **kwargs):
-    """Estimate GPU time needed based on steps and resolution."""
+    """Estimate GPU time needed based on steps and resolution.
+    Measured: ~15s for 30 steps at 1024x1024 with SDPA backend + content screening.
+    """
     pixels = height * width
-    base = 60
-    step_time = 3.5 if pixels <= 1024 * 1024 else 6.0
+    base = 20  # content screening + text encoding overhead
+    step_time = 0.5 if pixels <= 1024 * 1024 else 1.0
     return min(300, base + int(steps * step_time))
 
 
@@ -77,7 +78,7 @@ CSS = """
 .dark .gradio-container { color: var(--body-text-color); }
 """
 
-with gr.Blocks(theme=gr.themes.Citrus(), css=CSS) as demo:
+with gr.Blocks() as demo:
     gr.Markdown("# 🎨 Mage-Flow-Base Text-to-Image")
     gr.Markdown(
         "Generate images with [microsoft/Mage-Flow-Base](https://huggingface.co/microsoft/Mage-Flow-Base), "
@@ -142,4 +143,4 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=CSS) as demo:
         api_name="generate",
     )
 
-demo.launch(mcp_server=True)
+demo.launch(mcp_server=True, theme=gr.themes.Citrus(), css=CSS)