tensorlake
/

speaker-diarization-3.1

Automatic Speech Recognition

pyannote-audio-pipeline

speaker-diarization

speaker-change-detection

voice-activity-detection

overlapped-speech-detection

Inference Endpoints

Model card Files Files and versions Community

rishiraj commited on Jul 25

Commit

37c8107

•

1 Parent(s): d585bff

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -28,7 +28,7 @@ import torch
 class EndpointHandler:
     def __init__(self, path=""):
         # initialize pretrained pipeline
-        self._pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1")
         # send pipeline to GPU if available
         if torch.cuda.is_available():

 class EndpointHandler:
     def __init__(self, path=""):
         # initialize pretrained pipeline
+        self._pipeline = Pipeline.from_pretrained("tensorlake/speaker-diarization-3.1")
         # send pipeline to GPU if available
         if torch.cuda.is_available():