duration fixed

app.py

@@ -147,7 +147,7 @@ def update_required_time(steps, max_duration):
     return get_required_time_string(steps, max_duration)
 
 
-def generate_scene(prompt_text, steps, image_paths, audio_file_path, tea_cache_l1_thresh, max_duration = 2, session_id = None):
+def generate_scene(prompt_text, steps, image_paths, audio_file_path, tea_cache_l1_thresh, max_duration = 3, session_id = None):
 
     print(image_paths)
     prompt_text_check = (prompt_text or "").strip()
@@ -160,9 +160,9 @@ def generate_scene(prompt_text, steps, image_paths, audio_file_path, tea_cache_l
     return run_pipeline(prompt_text, steps, image_paths, audio_file_path, tea_cache_l1_thresh, max_duration, session_id)
 
 
-def generate_example(prompt_text, steps, image_paths, audio_file_path, tea_cache_l1_thresh = 0.0, max_duration = 4, session_id = None):
+def generate_example(prompt_text, steps, image_paths, audio_file_path):
 
-    return run_pipeline(prompt_text, steps, image_paths, audio_file_path, tea_cache_l1_thresh, max_duration, session_id)
+    return run_pipeline(prompt_text, steps, image_paths, audio_file_path)
 
 def upload_inductor_cache_to_hub(
     repo_id: str,
@@ -216,7 +216,7 @@ def upload_inductor_cache_to_hub(
 
 
 @spaces.GPU(duration=get_duration)
-def run_pipeline(prompt_text, steps, image_paths, audio_file_path, tea_cache_l1_thresh = 0.0, max_duration = 2, session_id = None):
+def run_pipeline(prompt_text, steps, image_paths, audio_file_path, tea_cache_l1_thresh = 0.0, max_duration = 3, session_id = None):
 
     if session_id is None:
         session_id = uuid.uuid4().hex
@@ -348,9 +348,9 @@ with gr.Blocks(css=css) as demo:
         gr.Markdown("**SETTINGS**")
 
         default_steps = 10
-        default_max_duration = 2
+        default_max_duration = 3
 
-        max_duration = gr.Slider(minimum=2, maximum=5, value=default_max_duration, step=1, label="Max Duration")
+        max_duration = gr.Slider(minimum=2, maximum=4, value=default_max_duration, step=1, label="Max Duration")
         steps_input = gr.Slider(minimum=10, maximum=50, value=default_steps, step=5, label="Diffusion Steps")
         tea_cache_l1_thresh = gr.Slider(minimum=0.0, maximum=1.0, value=0.0, step=0.01, label="Cache", visible=False)