fix_readme

main_pipeline.py

@@ -42,19 +42,20 @@ def save_speaker_audios(segments, denoised_audio_path, out_folder='out', out_f=4
     return out_wav_paths[:10]
 
 
-def main_pipeline(audio_path):
+def main_pipeline(audio_path, out_folder='out'):
     device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
     denoised_audio_path = denoise(audio_path, device)
     segments = diarization(denoised_audio_path)
-    denoised_audio_paths = save_speaker_audios(segments, denoised_audio_path, out_folder='out')
+    denoised_audio_paths = save_speaker_audios(segments, denoised_audio_path, out_folder)
     return denoised_audio_path, denoised_audio_paths
 
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
     parser.add_argument('--audio-path', default='dialog.mp3', help='Path to audio')
+    parser.add_argument('--out-folder-path', default='dialog.mp3', help='Path to audio')
     opt = parser.parse_args()
 
     for _ in tqdm(range(10)):
-        main_pipeline(audio_path=opt.audio_path)
+        main_pipeline(audio_path=opt.audio_path, out_folder=opt.out_folder_path)