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

b92adf3

•

1 Parent(s): 51657dd

Update handler.py

Files changed (1) hide show

handler.py +1 -2

handler.py CHANGED Viewed

@@ -6,11 +6,10 @@ import torch
 SAMPLE_RATE = 16000
 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]:

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