From a6de62f1217927669c4e665d05a3c34a7616020c Mon Sep 17 00:00:00 2001 From: Rishikanth Chandrasekaran Date: Tue, 27 May 2025 11:55:52 -0700 Subject: [PATCH] Addresses #119 - update whisperx syntax to match latest version --- diarize.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/diarize.py b/diarize.py index c3aa14b..919ea4f 100644 --- a/diarize.py +++ b/diarize.py @@ -37,7 +37,7 @@ def diarize_transcript(audio_file, transcript, device="cpu", model_name="pyannot print("Could not import torch for CUDA memory management") print(f"Loading diarization model: {model_name}") - diarize_model = whisperx.DiarizationPipeline( + diarize_model = whisperx.diarize.DiarizationPipeline( model_name=model_name, device=device, use_auth_token=hf_token if hf_token else None @@ -127,4 +127,4 @@ def diarize_transcript(audio_file, transcript, device="cpu", model_name="pyannot # Fallback: Assign "unknown" to original segments for segment in transcript["segments"]: segment["speaker"] = "unknown" - return transcript \ No newline at end of file + return transcript