From 2efa136114a3ad677a76c8e7b9e75008b3bb4f60 Mon Sep 17 00:00:00 2001 From: Max Bain <36994049+m-bain@users.noreply.github.com> Date: Mon, 8 May 2023 17:20:38 +0100 Subject: [PATCH] update python usage example --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index a6d400c..9d61d9b 100644 --- a/README.md +++ b/README.md @@ -176,10 +176,10 @@ print(result["segments"]) # after alignment diarize_model = whisperx.DiarizationPipeline(use_auth_token=YOUR_HF_TOKEN, device=device) # add min/max number of speakers if known -diarize_segments = diarize_model(input_audio_path) -# diarize_model(input_audio_path, min_speakers=min_speakers, max_speakers=max_speakers) +diarize_segments = diarize_model(audio_file) +# diarize_model(audio_file, min_speakers=min_speakers, max_speakers=max_speakers) -result = assign_word_speakers(diarize_segments, result) +result = whisperx.assign_word_speakers(diarize_segments, result) print(diarize_segments) print(result["segments"]) # segments are now assigned speaker IDs ```