Forensic-Noise-Classifier / utils /speaker_diarization.py
Knight-coderr's picture
Upload 2 files
335eff1 verified
raw
history blame contribute delete
298 Bytes
from pyannote.audio import Pipeline
import os
hf_token = os.environ.get("HF_TOKEN")
diarization_pipeline = Pipeline.from_pretrained(
"pyannote/speaker-diarization-3.1",
use_auth_token=hf_token
)
def diarize_speakers(audio_path):
return diarization_pipeline(audio_path)