back one eol

app.py

@@ -18,11 +18,11 @@ MODEL_CACHE = {
 datamix = AudioMixer()
 
 
-def gradio_TSE(input_audio_path, enroll_audio_path1, enroll_audio_path2, audio_type, model_select):
+def gradio_TSE(input_audio_path, enroll_audio_path1, audio_type, model_select):
     print(f"模型选择: {model_select}")
     print(f"User uploaded audio path: {input_audio_path}")
     print(f"User enroll audio path:  {enroll_audio_path1}")
-    print(f"User enroll audio path:  {enroll_audio_path2}")
+    # print(f"User enroll audio path:  {enroll_audio_path2}")
 
     # if model_select == "base_model":
     #     cfg_path = "config/config_base.yaml"
@@ -39,8 +39,8 @@ def gradio_TSE(input_audio_path, enroll_audio_path1, enroll_audio_path2, audio_t
     eol_info = sf.info(enroll_audio_path1)
     print(f"采样率: {eol_info.samplerate} Hz")
 
-    eol_info = sf.info(enroll_audio_path2)
-    print(f"采样率: {eol_info.samplerate} Hz")
+    # eol_info = sf.info(enroll_audio_path2)
+    # print(f"采样率: {eol_info.samplerate} Hz")
 
     input_deal_wav = fix_audio_format(input_audio_path)
     input_deal_wav_path = "deal_input.wav"
@@ -58,10 +58,10 @@ def gradio_TSE(input_audio_path, enroll_audio_path1, enroll_audio_path2, audio_t
     #     eol_wav2 = "eol2.wav"
     #     sf.write(eol_wav2, enroll_wav2, 16000)
     # else:
-    enroll_wav2 = fix_audio_format(enroll_audio_path2)
-    print("成功导入 enroll_wav2，长度:", len(enroll_wav2))
-    eol_wav2 = "eol2.wav"
-    sf.write(eol_wav2, enroll_wav2, 16000)
+    # enroll_wav2 = fix_audio_format(enroll_audio_path2)
+    # print("成功导入 enroll_wav2，长度:", len(enroll_wav2))
+    # eol_wav2 = "eol2.wav"
+    # sf.write(eol_wav2, enroll_wav2, 16000)
 
     if audio_type == "clean":
         noise_folder_test = "noises/" 
@@ -75,9 +75,9 @@ def gradio_TSE(input_audio_path, enroll_audio_path1, enroll_audio_path2, audio_t
 
     est_path1 = inter.run_inference(mix_path, eol_wav1)
 
-    est_path2 = inter.run_inference(mix_path, eol_wav2)
+    # est_path2 = inter.run_inference(mix_path, eol_wav2)
 
-    return mix_path,est_path1,est_path2
+    return mix_path,est_path1
 
 
 
@@ -104,7 +104,7 @@ with gr.Blocks() as demo:
     with gr.Row():
 
         enroll_audio1 = gr.Audio(label="Upload your first enroll audio", type="filepath")
-        enroll_audio2 = gr.Audio(label="Upload your second enroll audio to compare", type="filepath")
+        # enroll_audio2 = gr.Audio(label="Upload your second enroll audio to compare", type="filepath")
     
     with gr.Row():
 
@@ -112,13 +112,13 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         extracted_audio_output1 = gr.Audio(label="First enroll extracted target speaker audio", type="filepath")
-        extracted_audio_output2 = gr.Audio(label="Second enroll extracted target speaker audio", type="filepath")
+        # extracted_audio_output2 = gr.Audio(label="Second enroll extracted target speaker audio", type="filepath")
 
     convert_button = gr.Button("Extract")
     convert_button.click(
         fn=gradio_TSE,
-        inputs=[input_audio, enroll_audio1, enroll_audio2, audio_type, model_select],
-        outputs=[noisy_audio_output, extracted_audio_output1, extracted_audio_output2]
+        inputs=[input_audio, enroll_audio1, audio_type, model_select],
+        outputs=[noisy_audio_output, extracted_audio_output1]
     )
 
 if __name__ == "__main__":