jguerrisi commited on
Commit
a298116
·
verified ·
1 Parent(s): e4c8a5f

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +20 -19
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
- @lru_cache(maxsize=10)
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
+