Set prerendered examples, set max_size queue

app.py

@@ -229,6 +229,15 @@ def infer(prompt, cond_image_path, cond_audio_path_spk1, cond_audio_path_spk2, s
             except Exception as e:
                 print(f"[WARNING] Could not remove {f}: {e}")  
 
+def load_prerendered_examples(prompt, cond_image_path, cond_audio_path_spk1, cond_audio_path_spk2, sample_steps):
+    output_video = None
+    
+    if cond_image_path == "examples/single/single1.png":
+        output_video = "examples/results/multitalk_single_example_1.mp4"
+    elif cond_image_path == "examples/multi/3/multi3.png":
+        output_video = "examples/results/multitalk_multi_example_2.mp4"
+
+    return output_video
 
 with gr.Blocks(title="MultiTalk Inference") as demo:
     gr.Markdown("## 🎤 Meigen MultiTalk Inference Demo")
@@ -290,7 +299,7 @@ with gr.Blocks(title="MultiTalk Inference") as demo:
                     ["A woman sings passionately in a dimly lit studio.", "examples/single/single1.png", "examples/single/1.wav", None, 12],
                     ["In a cozy recording studio, a man and a woman are singing together. The man, with tousled brown hair, stands to the left, wearing a light green button-down shirt. His gaze is directed towards the woman, who is smiling warmly. She, with wavy dark hair, is dressed in a black floral dress and stands to the right, her eyes closed in enjoyment. Between them is a professional microphone, capturing their harmonious voices. The background features wooden panels and various audio equipment, creating an intimate and focused atmosphere. The lighting is soft and warm, highlighting their expressions and the intimate setting. A medium shot captures their interaction closely.", "examples/multi/3/multi3.png", "examples/multi/3/1-man.WAV", "examples/multi/3/1-woman.WAV", 12],
                 ],
-                fn=infer,
+                fn=load_prerendered_examples,
                 inputs = [prompt_input, image_input, audio_input_spk1, audio_input_spk2, sample_steps],
                 outputs=output_video,
                 cache_examples = True,
@@ -303,4 +312,4 @@ with gr.Blocks(title="MultiTalk Inference") as demo:
         outputs=output_video
     )
 
-demo.launch(ssr_mode=False, show_error=True, show_api=False)
+demo.queue(max_size=2).launch(ssr_mode=False, show_error=True, show_api=False)