Update model.py
Browse files
model.py
CHANGED
@@ -1008,6 +1008,7 @@ def _get_dolphin_ctc_models(repo_id: str, decoding_method: str, num_active_paths
|
|
1008 |
"csukuangfj/sherpa-onnx-dolphin-small-ctc-multi-lang-int8-2025-04-02",
|
1009 |
"csukuangfj/sherpa-onnx-dolphin-base-ctc-multi-lang-2025-04-02",
|
1010 |
"csukuangfj/sherpa-onnx-dolphin-small-ctc-multi-lang-2025-04-02",
|
|
|
1011 |
], repo_id
|
1012 |
|
1013 |
if repo_id in [
|
@@ -1029,6 +1030,14 @@ def _get_dolphin_ctc_models(repo_id: str, decoding_method: str, num_active_paths
|
|
1029 |
subfolder=".",
|
1030 |
)
|
1031 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1032 |
recognizer = sherpa_onnx.OfflineRecognizer.from_dolphin_ctc(
|
1033 |
tokens=tokens,
|
1034 |
model=nn_model,
|
|
|
1008 |
"csukuangfj/sherpa-onnx-dolphin-small-ctc-multi-lang-int8-2025-04-02",
|
1009 |
"csukuangfj/sherpa-onnx-dolphin-base-ctc-multi-lang-2025-04-02",
|
1010 |
"csukuangfj/sherpa-onnx-dolphin-small-ctc-multi-lang-2025-04-02",
|
1011 |
+
"csukuangfj/sherpa-onnx-nemo-canary-180m-flash-en-es-de-fr-int8",
|
1012 |
], repo_id
|
1013 |
|
1014 |
if repo_id in [
|
|
|
1030 |
subfolder=".",
|
1031 |
)
|
1032 |
|
1033 |
+
if "canary" in repo_id:
|
1034 |
+
recognizer = sherpa_onnx.OfflineRecognizer.from_nemo_ctc(
|
1035 |
+
tokens=tokens,
|
1036 |
+
model=nn_model,
|
1037 |
+
num_threads=2,
|
1038 |
+
language="es", # <-- Force Spanish here
|
1039 |
+
)
|
1040 |
+
else:
|
1041 |
recognizer = sherpa_onnx.OfflineRecognizer.from_dolphin_ctc(
|
1042 |
tokens=tokens,
|
1043 |
model=nn_model,
|