fix(zerogpu): use dynamic *args in get_duration and disable experimental ssr

app.py

@@ -473,14 +473,19 @@ def _fit_keyframe(image_input, target_width: int, target_height: int) -> Image.I
     return img
 
 
-def get_duration(prompt, negative_prompt, input_image, nsfw_base_choice, nsfw_base_strength, canvas, num_frames, fps, steps, *a, **k):
-    steps = int(steps)
-    num_frames = int(num_frames)
-    h, w = CANVASES.get(canvas, (480, 832))
-    pixels_per_frame = h * w
-    total_tokens = (pixels_per_frame / 256) * (num_frames / 4)
-    estimated = int(steps * (total_tokens * 0.00012) + 20)
-    return max(60, min(MAX_GPU_DURATION, estimated))
+def get_duration(*args, **kwargs):
+    try:
+        # _generate_video_gpu receives: (prompt, negative_prompt, image_input, height, width, num_frames, steps, ...)
+        height = kwargs.get("height", args[3] if len(args) > 3 else 480)
+        width = kwargs.get("width", args[4] if len(args) > 4 else 832)
+        num_frames = kwargs.get("num_frames", args[5] if len(args) > 5 else 81)
+        steps = kwargs.get("steps", args[6] if len(args) > 6 else 20)
+        pixels_per_frame = int(height) * int(width)
+        total_tokens = (pixels_per_frame / 256) * (int(num_frames) / 4)
+        estimated = int(int(steps) * (total_tokens * 0.00012) + 20)
+        return max(60, min(MAX_GPU_DURATION, estimated))
+    except Exception:
+        return 180
 
 
 @spaces.GPU(duration=get_duration, size=GPU_SIZE)
@@ -1051,4 +1056,4 @@ with gr.Blocks(css=custom_css, title="Wan 2.2 Video Studio") as app:
     )
 
 if __name__ == "__main__":
-    app.launch(show_api=False, show_error=True, allowed_paths=[OUTPUT_DIR])
+    app.launch(ssr=False, show_api=False, show_error=True, allowed_paths=[OUTPUT_DIR])