Update app.py

app.py

@@ -606,7 +606,35 @@ def on_highres_toggle(first_image, last_image, high_res):
     return gr.update(value=w), gr.update(value=h)
 
 
-def get_gpu_duration(gpu_duration: float):
+def get_gpu_duration(
+    first_image,
+    last_image,
+    input_audio,
+    prompt: str,
+    duration: float,
+    gpu_duration: float,
+    enhance_prompt: bool = True,
+    seed: int = 42,
+    randomize_seed: bool = True,
+    height: int = 1024,
+    width: int = 1536,
+    pose_strength: float = 0.0,
+    general_strength: float = 0.0,
+    motion_strength: float = 0.0,
+    dreamlay_strength: float = 0.0,
+    mself_strength: float = 0.0,
+    dramatic_strength: float = 0.0,
+    fluid_strength: float = 0.0,
+    liquid_strength: float = 0.0,
+    demopose_strength: float = 0.0,
+    voice_strength: float = 0.0,
+    realism_strength: float = 0.0,
+    transition_strength: float = 0.0,
+    physics_strength: float = 0.0,
+    reasoning_strength: float = 0.0,
+    twostep_strength: float = 0.0,
+    progress=None,
+):
     return int(gpu_duration)
 
 @spaces.GPU(duration=get_gpu_duration)