example test

app.py

@@ -114,7 +114,7 @@ def restore_inductor_cache_from_hub(repo_id: str, filename: str = "torch_compile
 # restore_inductor_cache_from_hub("alexnasa/humo-compiled")
 
 
-def get_duration(prompt_text, steps, image_file, audio_file_path, tea_cache_l1_thresh, max_duration, session_id):
+def get_duration(prompt_text, steps, image_file, audio_file_path, tea_cache_l1_thresh, max_duration, session_id, progress):
 
     return calculate_required_time(steps, max_duration)
 
@@ -148,7 +148,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 = 2, session_id = None, progress=gr.Progress(track_tqdm=True)):
 
     print(image_paths)
     prompt_text_check = (prompt_text or "").strip()
@@ -158,9 +158,12 @@ def generate_scene(prompt_text, steps, image_paths, audio_file_path, tea_cache_l
     if not audio_file_path and not image_paths:
         raise gr.Error("Please provide a reference image or a lipsync audio.")
     
-    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, tea_cache_l1_thresh, max_duration, session_id, progress)
 
 
+def generate_example(prompt_text, steps, image_paths, audio_file_path, tea_cache_l1_thresh = 0.0, max_duration = 5, session_id = None, progress=gr.Progress(track_tqdm=True)):
+
+    return run_pipeline(prompt_text, steps, image_paths, audio_file_path, tea_cache_l1_thresh, max_duration, session_id, progress)
 
 def upload_inductor_cache_to_hub(
     repo_id: str,
@@ -214,7 +217,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 = 2, session_id = None, progress=gr.Progress(track_tqdm=True)):
 
     if session_id is None:
         session_id = uuid.uuid4().hex
@@ -416,7 +419,7 @@ with gr.Blocks(css=css) as demo:
             ],
             inputs=[prompt_tb, steps_input, img_input, audio_input],
             outputs=[video_output],
-            fn=run_pipeline,
+            fn=generate_example,
             cache_examples=True,
         )
         max_duration.change(update_required_time, [steps_input, max_duration], time_required)