Spaces:

k2-fsa
/

automatic-speech-recognition

Running

csukuangfj commited on Aug 7, 2023

Commit

5cb3bb4

1 Parent(s): 60acde8

minor fixes

Files changed (1) hide show

model.py CHANGED Viewed

@@ -275,7 +275,7 @@ def _get_whisper_model(
 ) -> sherpa_onnx.OfflineRecognizer:
     print(f"get {repo_id}")
     assert repo_id in ("tiny.en", "base.en", "small.en", "medium.en"), repo_id
-    name = repo_id
     full_repo_id = "csukuangfj/sherpa-onnx-whisper-" + name
     encoder = _get_nn_model_filename(
         repo_id=full_repo_id,

 ) -> sherpa_onnx.OfflineRecognizer:
     print(f"get {repo_id}")
     assert repo_id in ("tiny.en", "base.en", "small.en", "medium.en"), repo_id
+    name = repo_id.split("-")[1]
     full_repo_id = "csukuangfj/sherpa-onnx-whisper-" + name
     encoder = _get_nn_model_filename(
         repo_id=full_repo_id,