fix eol2 bug

app.py

@@ -39,18 +39,18 @@ def gradio_TSE(input_audio_path, enroll_audio_path1, enroll_audio_path2, audio_t
     eol_wav1 = "eol1.wav"
     sf.write(eol_wav1, enroll_wav1, 16000)
 
-    if len(enroll_wav1) > 4 * 16000:
-        middle_start = (len(enroll_wav1) - 3 * 16000) // 2
-        middle_end = middle_start + 3 * 16000
-        enroll_wav2 = enroll_wav1[middle_start:middle_end]
-        print("成功提取 enroll_wav2，长度:", len(enroll_wav2))
-        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)
+    # if len(enroll_wav1) > 4 * 16000:
+    #     middle_start = (len(enroll_wav1) - 3 * 16000) // 2
+    #     middle_end = middle_start + 3 * 16000
+    #     enroll_wav2 = enroll_wav1[middle_start:middle_end]
+    #     print("成功提取 enroll_wav2，长度:", len(enroll_wav2))
+    #     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)
 
     if audio_type == "clean":
         noise_folder_test = "noises/" 

decode.py

@@ -64,7 +64,7 @@ if __name__ == "__main__":
     mix_path = "test_output_mixture.wav"
     enroll_path = "test_mix.wav"
     out_path = "temp_output.wav"
-    out_wav = pipeline.run_inference(mix_path, enroll_path, out_path)
+    out_wav = pipeline.run_inference(mix_path, enroll_path)
     print("Done:", out_wav)