Spaces:

k2-fsa
/

automatic-speech-recognition

Running

csukuangfj commited on Dec 14, 2023

Commit

5f1b2ca

1 Parent(s): 463d335

fix whisper

Files changed (1) hide show

model.py CHANGED Viewed

@@ -330,13 +330,13 @@ def _get_whisper_model(
     full_repo_id = "csukuangfj/sherpa-onnx-whisper-" + name
     encoder = _get_nn_model_filename(
         repo_id=full_repo_id,
-        filename=f"{name}-encoder.int8.ort",
         subfolder=".",
     )
     decoder = _get_nn_model_filename(
         repo_id=full_repo_id,
-        filename=f"{name}-decoder.int8.ort",
         subfolder=".",
     )

     full_repo_id = "csukuangfj/sherpa-onnx-whisper-" + name
     encoder = _get_nn_model_filename(
         repo_id=full_repo_id,
+        filename=f"{name}-encoder.int8.onnx",
         subfolder=".",
     )
     decoder = _get_nn_model_filename(
         repo_id=full_repo_id,
+        filename=f"{name}-decoder.int8.onnx",
         subfolder=".",
     )