Update app.py

app.py

@@ -229,11 +229,20 @@ def update_time_required(max_duration_s, rc_str):
     
     return gr.update(value=f"⌚ Zero GPU Required: ~{duration_s}.0s ({duration_m:.1f} mins)")
 
-def get_duration(input_video, max_duration_s, edited_frame, rc_bool, session_id, progress):
-
-    return calculate_time_required(max_duration_s, rc_bool)
+def log_slider_change():
+    print("slider value changed")
 
-@spaces.GPU(duration=120)
+def get_duration(input_video, max_duration_s, edited_frame, rc_bool, session_id, progress):
+    if max_duration_s == 2:
+        return 120
+    elif max_duration_s == 4:
+        return 180
+    elif max_duration_s == 6:
+        return 240
+    elif max_duration_s == 8:
+        return 300
+
+@spaces.GPU(duration=get_duration)
 def _animate(input_video, max_duration_s, edited_frame, rc_bool, session_id = None, progress=gr.Progress(track_tqdm=True),):
 
     if session_id is None:
@@ -481,7 +490,7 @@ with gr.Blocks(css=css, title="Wan 2.2 Animate --replace", theme=gr.themes.Ocean
                 </div>
                 """)
                 input_video = gr.Video(label="Input Video", height=512)
-                max_duration_slider = gr.Slider(2, 8, 2, step=2, label="Max Duration", visible=False)
+                max_duration_slider = gr.Slider(2, 4, 2, step=2, label="Max Duration", visible=False)
 
                 gr.Examples(
                     examples=[
@@ -604,7 +613,7 @@ with gr.Blocks(css=css, title="Wan 2.2 Animate --replace", theme=gr.themes.Ocean
                 )
 
     action_button.click(fn=animate_scene, inputs=[input_video, max_duration_slider, edited_frame, replace_character_string, session_state], outputs=[output_video, pose_video, bg_video, mask_video, face_video])
-    max_duration_slider.change(update_time_required, inputs=[max_duration_slider, replace_character_string], outputs=[time_required])
+    max_duration_slider.change(log_slider_change)
     replace_character_string.change(update_time_required, inputs=[max_duration_slider, replace_character_string], outputs=[time_required])
                    
 if __name__ == "__main__":