philschmid
/

pyannote-speaker-diarization-endpoint

Model card Files Files and versions Community

philschmid HF staff commited on Nov 8, 2022

Commit

5e765ab

•

1 Parent(s): e82923c

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

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

 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]: