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,
|