Update model.py
Browse files
model.py
CHANGED
@@ -1378,6 +1378,25 @@ def _get_sherpa_onnx_nemo_transducer_models(
|
|
1378 |
|
1379 |
return recognizer
|
1380 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1381 |
|
1382 |
@lru_cache(maxsize=10)
|
1383 |
def _get_sherpa_onnx_nemo_ctc_models(
|
@@ -2197,22 +2216,4 @@ language_to_models = {
|
|
2197 |
"Vietnamese": list(vietnamese_models.keys()),
|
2198 |
}
|
2199 |
|
2200 |
-
|
2201 |
-
def _get_nemo_ctc_models(repo_id: str, decoding_method: str, num_active_paths: int):
|
2202 |
-
nn_model = _get_nn_model_filename(
|
2203 |
-
repo_id=repo_id,
|
2204 |
-
filename="model.int8.onnx",
|
2205 |
-
subfolder=".",
|
2206 |
-
)
|
2207 |
-
tokens = _get_token_filename(
|
2208 |
-
repo_id=repo_id,
|
2209 |
-
filename="tokens.txt",
|
2210 |
-
subfolder=".",
|
2211 |
-
)
|
2212 |
-
recognizer = sherpa_onnx.OfflineRecognizer.from_nemo_ctc(
|
2213 |
-
tokens=tokens,
|
2214 |
-
model=nn_model,
|
2215 |
-
num_threads=2,
|
2216 |
-
language="es", # Force Spanish
|
2217 |
-
)
|
2218 |
-
return recognizer
|
|
|
1378 |
|
1379 |
return recognizer
|
1380 |
|
1381 |
+
@lru_cache(maxsize=10)
|
1382 |
+
def _get_nemo_ctc_models(repo_id: str, decoding_method: str, num_active_paths: int):
|
1383 |
+
nn_model = _get_nn_model_filename(
|
1384 |
+
repo_id=repo_id,
|
1385 |
+
filename="model.int8.onnx",
|
1386 |
+
subfolder=".",
|
1387 |
+
)
|
1388 |
+
tokens = _get_token_filename(
|
1389 |
+
repo_id=repo_id,
|
1390 |
+
filename="tokens.txt",
|
1391 |
+
subfolder=".",
|
1392 |
+
)
|
1393 |
+
recognizer = sherpa_onnx.OfflineRecognizer.from_nemo_ctc(
|
1394 |
+
tokens=tokens,
|
1395 |
+
model=nn_model,
|
1396 |
+
num_threads=2,
|
1397 |
+
language="es", # Force Spanish
|
1398 |
+
)
|
1399 |
+
return recognizer
|
1400 |
|
1401 |
@lru_cache(maxsize=10)
|
1402 |
def _get_sherpa_onnx_nemo_ctc_models(
|
|
|
2216 |
"Vietnamese": list(vietnamese_models.keys()),
|
2217 |
}
|
2218 |
|
2219 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|