philschmid HF staff commited on
Commit
5e765ab
1 Parent(s): e82923c

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -10,7 +10,7 @@ SAMPLE_RATE = 16000
10
  class EndpointHandler():
11
  def __init__(self, path=""):
12
  # load the model
13
- self.pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization")
14
 
15
 
16
  def __call__(self, data: Dict[str, bytes]) -> Dict[str, str]:
 
10
  class EndpointHandler():
11
  def __init__(self, path=""):
12
  # load the model
13
+ self.pipeline = Pipeline.from_pretrained("philschmid/pyannote-speaker-diarization-endpoint")
14
 
15
 
16
  def __call__(self, data: Dict[str, bytes]) -> Dict[str, str]: