Update app.py

app.py

@@ -40,10 +40,6 @@ example_mapping = {
     "Example 3": {"audio": os.path.join(example_folder, "example3.wav"), "image": os.path.join(example_folder, "example3.png")},
 }
 
-def load_example(choice):
-    example = example_mapping.get(choice, {})
-    return example.get("audio", ""), example.get("image", "")
-
 @spaces.GPU
 def predict(prompt, upload_reference_audio, microphone_reference_audio, reference_audio_type):
     global result_dir
@@ -129,16 +125,21 @@ def main():
                             )
                     with gr.TabItem("Example"):
                         example_choice = gr.Dropdown(choices=example_choices, label="Choose an example")
-                        example_audio = gr.Audio(label="Example Audio", type="filepath")
-                        example_image = gr.Image(label="Example Image", type="filepath")
-                        
+                        example_audio_state = gr.State("")
+                        example_image_state = gr.State("")
+                        def load_example(choice):
+                            example = example_mapping.get(choice, {})
+                            audio_path = example.get("audio", "")
+                            image_path = example.get("image", "")
+                            return audio_path, image_path, audio_path, image_path  # First pair updates state, second updates UI
                         example_choice.change(
                             fn=load_example, 
                             inputs=[example_choice], 
-                            outputs=[upload_driven_audio, source_image]
+                            outputs=[example_audio_state, example_image_state, upload_driven_audio, source_image]
                         )
                         example_choice.change(set_upload, outputs=driven_audio_type)
 
+
     return interface