make diarization faster

This commit is contained in:
Dudu Asulin
2023-08-02 10:11:43 +03:00
committed by GitHub
parent d80b98601b
commit 8de0e2af51
3 changed files with 10 additions and 3 deletions

View File

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