sprt/main.py

11 lines
402 B
Python

# load the pipeline from Hugginface Hub
from pyannote.audio import Pipeline
pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization@2022.07")
# apply the pipeline to an audio file
diarization = pipeline("f24122a7f35de434d996.ogg", min_speakers=3, max_speakers=9)
# dump the diarization output to disk using RTTM format
with open("audio.rttm", "w") as rttm:
diarization.write_rttm(rttm)