Update app.py

app.py

@@ -58,8 +58,7 @@ def fn_clearvoice_tse(input_video):
     print(f'input_video: {input_video}')
     myClearVoice(input_path=input_video, online_write=True, output_path='path_to_output_videos_tse')
 
-    output_list = find_mp4_files('path_to_output_videos_tse/')
-    print(output_list)
+    output_list = find_mp4_files(f'path_to_output_videos_tse/AV_MossFormer2_TSE_16K/{input_video.split('.')[0]}/')
     
     return output_list
 
@@ -126,8 +125,8 @@ tse_demo = gr.Interface(
     article = ("<p style='text-align: center'><a href='https://arxiv.org/abs/2302.11824' target='_blank'>MossFormer: Pushing the Performance Limit of Monaural Speech Separation using Gated Single-Head Transformer with Convolution-Augmented Joint Self-Attentions</a> | <a href='https://github.com/alibabasglab/MossFormer' target='_blank'>Github Repo</a></p>"
               "<p style='text-align: center'><a href='https://arxiv.org/abs/2312.11825' target='_blank'>MossFormer2: Combining Transformer and RNN-Free Recurrent Network for Enhanced Time-Domain Monaural Speech Separation</a> | <a href='https://github.com/alibabasglab/MossFormer2' target='_blank'>Github Repo</a></p>"),
     examples = [
-        ['examples/female_female_speech.wav'],
-        ['examples/female_male_speech.wav'],
+        ['examples/001.mp4'],
+        ['examples/002.mp4'],
     ],
     cache_examples = True,
 )