denoise_diar_app

main_pipeline.py

@@ -22,7 +22,7 @@ def save_speaker_audios(segments, denoised_audio_path, out_folder='out', out_f=4
             start = int(r["start"] * out_f)
             end = int(r["end"] * out_f)
             # output_signal[start:end] = signal[start:end]
-            out_wav_paths.append(signal[start:end])
+            output_signal.append(signal[start:end])
 
         out_wav_path = f'{out_folder}/{label}.wav'
         sf.write(out_wav_path, np.concatenate(output_signal), out_f, 'PCM_24')
@@ -42,4 +42,4 @@ def main_pipeline(audio_path):
 
 
 if __name__ == '__main__':
-    main_pipeline('dialog.mp3')
+    main_pipeline('out.wav')