TenzinGayche
/

whisper-small-3

Automatic Speech Recognition

Inference Endpoints

Model card Files Files and versions Community

TenzinGayche commited on Sep 21, 2023

Commit

31ba694

•

1 Parent(s): 6212570

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -12,7 +12,7 @@ SAMPLE_RATE = 16000
 class EndpointHandler():
     def __init__(self, path=""):
         # load the model
-        self.pipe = pipeline(model="TenzinGayche/whisper-small-3",device='cuda')
     def __call__(self, data: Dict[str, bytes]) -> Dict[str, str]:

 class EndpointHandler():
     def __init__(self, path=""):
         # load the model
+        self.pipe = pipeline(model="TenzinGayche/whisper-small-3",chunk_length_s=30,device='cuda')
     def __call__(self, data: Dict[str, bytes]) -> Dict[str, str]: