chrissewell/whisperX
1
0
Fork 0
mirror of https://github.com/m-bain/whisperX.git synced 2025-07-01 18:17:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

more

Browse Source
This commit is contained in:
Dudu Asulin
2023-08-02 10:35:20 +03:00
committed by GitHub
parent da6ed83dc9
commit 577db33430
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
whisperx/transcribe.py
View File

@ -202,8 +202,7 @@ def cli():
results = [] results = []
diarize_model = DiarizationPipeline(use_auth_token=hf_token, device=device) diarize_model = DiarizationPipeline(use_auth_token=hf_token, device=device)
for result, input_audio_path in tmp_results: for result, input_audio_path in tmp_results:
audio = load_audio(input_audio_path) diarize_segments = diarize_model(input_audio_path, min_speakers=min_speakers, max_speakers=max_speakers)
diarize_segments = diarize_model(audio, min_speakers=min_speakers, max_speakers=max_speakers)
result = assign_word_speakers(diarize_segments, result) result = assign_word_speakers(diarize_segments, result)
results.append((result, input_audio_path)) results.append((result, input_audio_path))
# >> Write # >> Write