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