johnislarry commited on
Commit
214cf2e
1 Parent(s): 025df12

Update handler.py

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