Allow cold four-step edits to finish on ZeroGPU

app.py

@@ -114,7 +114,7 @@ def use_output_as_input(output_images):
 
 
 # --- Main Inference Function (with hardcoded negative prompt) ---
-@spaces.GPU(duration=60)
+@spaces.GPU(duration=180)
 def infer(
     image_1,
     image_2,
@@ -159,7 +159,8 @@ def infer(
         height, width = None, None
     print(
         f"Starting generation: seed={seed}, steps={num_inference_steps}, "
-        f"guidance={true_guidance_scale}, size={width}x{height}, inputs={len(pil_images)}"
+        f"guidance={true_guidance_scale}, size={width}x{height}, inputs={len(pil_images)}",
+        flush=True,
     )
 
     # Generate the image