zou8944 commited on
Commit
12ade36
·
verified ·
1 Parent(s): 3f7be94

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -2
handler.py CHANGED
@@ -11,8 +11,7 @@ SAMPLE_RATE = 16000
11
  class EndpointHandler():
12
  def __init__(self, path=""):
13
  # load the model
14
- model_path = hf_hub_download(repo_id="openai/whisper-large-v3", filename="model.bin")
15
- self.model = load_model(model_path)
16
 
17
  def __call__(self, data: Dict[str, bytes]) -> Dict[str, str]:
18
  """
 
11
  class EndpointHandler():
12
  def __init__(self, path=""):
13
  # load the model
14
+ self.model = whisper.load_model("large-v3")
 
15
 
16
  def __call__(self, data: Dict[str, bytes]) -> Dict[str, str]:
17
  """