srd4
/

my-openai-whisper-endpoint

Automatic Speech Recognition

endpoints-template

Inference Endpoints

Model card Files Files and versions Community

srd4 commited on Dec 21, 2023

Commit

52fd856

•

1 Parent(s): 4d5673d

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -9,7 +9,7 @@ SAMPLE_RATE = 16000
 class EndpointHandler():
     def __init__(self, path=""):
         # load the model
-        self.model = whisper.load_model("large")
     def __call__(self, data: Dict[str, bytes]) -> Dict[str, str]:

 class EndpointHandler():
     def __init__(self, path=""):
         # load the model
+        self.model = whisper.load_model("medium")
     def __call__(self, data: Dict[str, bytes]) -> Dict[str, str]: