Upload app.py with huggingface_hub

app.py

@@ -269,11 +269,19 @@ def paste_back(bg_pil, generated_patch, inp):
 # ----------------------------------------------------------------------------
 
 def _estimate_duration(bg, obj, cx, cy, scale, seed, ref_scale, steps, *a, **k):
+    # Measured ~12 s/step at 1024 when reference guidance is on (CFG doubles the
+    # forward pass); ~half that when it is off. Plus fixed overhead for VAE /
+    # rembg / cold worker init.
     try:
         steps = int(steps)
     except Exception:
-        steps = 28
-    return min(230, 60 + int(steps * 4.5))
+        steps = 16
+    try:
+        ref_on = float(ref_scale) > 1.0
+    except Exception:
+        ref_on = True
+    per_step = 12.5 if ref_on else 6.5
+    return int(min(600, 45 + steps * per_step))
 
 
 @spaces.GPU(duration=_estimate_duration)
@@ -388,15 +396,15 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=CSS) as demo:
                 scale = gr.Slider(0.05, 0.9, value=0.35, step=0.01, label="Object size")
             with gr.Row():
                 ref_scale = gr.Slider(1.0, 5.0, value=2.0, step=0.1, label="Reference guidance scale")
-                steps = gr.Slider(15, 50, value=28, step=1, label="Inference steps")
+                steps = gr.Slider(12, 28, value=16, step=1, label="Inference steps")
                 seed = gr.Number(label="Seed", value=42, precision=0)
             rand_btn = gr.Button("🎲 Randomize seed")
 
         gr.Examples(
             examples=[
-                ["examples/bg_landscape.jpg", "examples/obj_ducks.jpg", 0.55, 0.70, 0.28, 42, 2.0, 28],
-                ["examples/bg_tent.jpg", "examples/obj_dog.jpg", 0.45, 0.68, 0.30, 7, 2.0, 28],
-                ["examples/bg_beach.jpg", "examples/obj_cake.jpg", 0.50, 0.72, 0.22, 123, 2.5, 30],
+                ["examples/bg_landscape.jpg", "examples/obj_ducks.jpg", 0.55, 0.70, 0.28, 42, 2.0, 16],
+                ["examples/bg_tent.jpg", "examples/obj_dog.jpg", 0.45, 0.68, 0.30, 7, 2.0, 16],
+                ["examples/bg_beach.jpg", "examples/obj_cake.jpg", 0.50, 0.72, 0.22, 123, 2.5, 16],
             ],
             inputs=[bg_input, obj_input, cx, cy, scale, seed, ref_scale, steps],
             outputs=[out_result, out_preview],