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