Update app.py

app.py

@@ -635,32 +635,44 @@ def on_highres_toggle(first_image, last_image, high_res):
     return gr.update(value=w), gr.update(value=h)
 
 
-def get_duration(
+def get_gpu_duration(
     first_image,
     last_image,
     prompt: str,
     negative_prompt: str,
     duration: float,
-    seed: int,
-    randomize_seed: bool,
-    height: int,
-    width: int,
-    enhance_prompt: bool,
-    video_cfg_scale: float,
-    video_stg_scale: float,
-    video_rescale_scale: float,
-    video_a2v_scale: float,
-    audio_cfg_scale: float,
-    audio_stg_scale: float,
-    audio_rescale_scale: float,
-    audio_v2a_scale: float,
     gpu_duration: float,
+    enhance_prompt: bool = True,
+    randomize_seed: bool = True,
+    height: int = 1024,
+    width: int = 1536,
+    video_cfg_scale: float = 1.0,
+    video_stg_scale: float = 0.0,
+    video_rescale_scale: float = 0.45,
+    video_a2v_scale: float = 3.0,
+    audio_cfg_scale: float = 1.0,
+    audio_stg_scale: float = 0.0,
+    audio_rescale_scale: float = 1.0,
+    audio_v2a_scale: float = 3.0,
+    distilled_strength: float = 0.0,
+    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,
     progress=None,
 ) -> int:
     return int(gpu_duration)
 
 
-@spaces.GPU(duration=get_duration)
+@spaces.GPU(duration=get_gpu_duration)
 @torch.inference_mode()
 def generate_video(
     first_image,
@@ -668,19 +680,32 @@ def generate_video(
     prompt: str,
     negative_prompt: str,
     duration: float,
-    seed: int,
-    randomize_seed: bool,
-    height: int,
-    width: int,
-    enhance_prompt: bool,
-    video_cfg_scale: float,
-    video_stg_scale: float,
-    video_rescale_scale: float,
-    video_a2v_scale: float,
-    audio_cfg_scale: float,
-    audio_stg_scale: float,
-    audio_rescale_scale: float,
-    audio_v2a_scale: float,
+    gpu_duration: float,
+    enhance_prompt: bool = True,
+    randomize_seed: bool = True,
+    height: int = 1024,
+    width: int = 1536,
+    video_cfg_scale: float = 1.0,
+    video_stg_scale: float = 0.0,
+    video_rescale_scale: float = 0.45,
+    video_a2v_scale: float = 3.0,
+    audio_cfg_scale: float = 1.0,
+    audio_stg_scale: float = 0.0,
+    audio_rescale_scale: float = 1.0,
+    audio_v2a_scale: float = 3.0,
+    distilled_strength: float = 0.0,
+    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,
     progress=gr.Progress(track_tqdm=True),
 ):
     try: