johnislarry
/

cloned-pyannote-speaker-diarization-endpoint

Voice Activity Detection

pyannote-audio-pipeline

speaker-diarization

speaker-change-detection

overlapped-speech-detection

Model card Files Files and versions

johnislarry commited on Apr 20, 2023

Commit

214cf2e

·

1 Parent(s): 025df12

Update handler.py

Files changed (1) hide show

handler.py +1 -2

handler.py CHANGED Viewed

@@ -10,8 +10,7 @@ SAMPLE_RATE = 16000
 class EndpointHandler():
     def __init__(self, path=""):
         # load the model
-        self.pipeline = Pipeline.from_pretrained("philschmid/pyannote-speaker-diarization-endpoint")
     def __call__(self, data: Dict[str, bytes]) -> Dict[str, str]:
         """

 class EndpointHandler():
     def __init__(self, path=""):
         # load the model
+        self.pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization@2.1")
     def __call__(self, data: Dict[str, bytes]) -> Dict[str, str]:
         """