Update model.py
Browse files
model.py
CHANGED
|
@@ -1031,18 +1031,18 @@ def _get_dolphin_ctc_models(repo_id: str, decoding_method: str, num_active_paths
|
|
| 1031 |
)
|
| 1032 |
|
| 1033 |
if "canary" in repo_id:
|
| 1034 |
-
|
| 1035 |
tokens=tokens,
|
| 1036 |
model=nn_model,
|
| 1037 |
num_threads=2,
|
| 1038 |
language="es", # <-- Force Spanish here
|
| 1039 |
-
|
| 1040 |
else:
|
| 1041 |
-
|
| 1042 |
tokens=tokens,
|
| 1043 |
model=nn_model,
|
| 1044 |
num_threads=2,
|
| 1045 |
-
|
| 1046 |
|
| 1047 |
return recognizer
|
| 1048 |
|
|
|
|
| 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,
|
| 1044 |
num_threads=2,
|
| 1045 |
+
)
|
| 1046 |
|
| 1047 |
return recognizer
|
| 1048 |
|