from pyannote.audio import Pipeline

diarization_pipeline = Pipeline.from_pretrained(
      "pyannote/speaker-diarization-3.1"
)

def diarize_speakers(audio_path):
    return diarization_pipeline(audio_path)