from pyannote.audio import Pipeline

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

def diarize_speakers(audio_path):
    return diarization_pipeline(audio_path)