vlad.os/whisperx-rocm-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

doc: fix diarize import in example script (#1192)

Browse Source
This commit is contained in:
Adrian Wan
2025-10-09 16:27:07 +09:00
committed by GitHub
parent c266ac5459
commit 2663f2edb5
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+2 -1
View File
@@ -166,6 +166,7 @@ See more examples in other languages [here](EXAMPLES.md).
```python ```python
import whisperx import whisperx
import gc import gc
from whisperx.diarize import DiarizationPipeline
device = "cuda" device = "cuda"
audio_file = "audio.mp3" audio_file = "audio.mp3"
@@ -196,7 +197,7 @@ print(result["segments"]) # after alignment
# import gc; import torch; gc.collect(); torch.cuda.empty_cache(); del model_a # import gc; import torch; gc.collect(); torch.cuda.empty_cache(); del model_a
# 3. Assign speaker labels # 3. Assign speaker labels
diarize_model = whisperx.diarize.DiarizationPipeline(use_auth_token=YOUR_HF_TOKEN, device=device) diarize_model = DiarizationPipeline(use_auth_token=YOUR_HF_TOKEN, device=device)
# add min/max number of speakers if known # add min/max number of speakers if known
diarize_segments = diarize_model(audio) diarize_segments = diarize_model(audio)