Skip to content

Commit 668909c

Browse files
committed
Addresses #119 - update whisperx syntax to match latest version
1 parent 185db58 commit 668909c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

diarize.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def diarize_transcript(audio_file, transcript, device="cpu", model_name="pyannot
3737
print("Could not import torch for CUDA memory management")
3838

3939
print(f"Loading diarization model: {model_name}")
40-
diarize_model = whisperx.DiarizationPipeline(
40+
diarize_model = whisperx.diarize.DiarizationPipeline(
4141
model_name=model_name,
4242
device=device,
4343
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
127127
# Fallback: Assign "unknown" to original segments
128128
for segment in transcript["segments"]:
129129
segment["speaker"] = "unknown"
130-
return transcript
130+
return transcript

0 commit comments

Comments
 (0)