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)