jguerrisi commited on
Commit
fdbcb84
·
verified ·
1 Parent(s): 264d985

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +9 -0
model.py CHANGED
@@ -1008,6 +1008,7 @@ def _get_dolphin_ctc_models(repo_id: str, decoding_method: str, num_active_paths
1008
  "csukuangfj/sherpa-onnx-dolphin-small-ctc-multi-lang-int8-2025-04-02",
1009
  "csukuangfj/sherpa-onnx-dolphin-base-ctc-multi-lang-2025-04-02",
1010
  "csukuangfj/sherpa-onnx-dolphin-small-ctc-multi-lang-2025-04-02",
 
1011
  ], repo_id
1012
 
1013
  if repo_id in [
@@ -1029,6 +1030,14 @@ def _get_dolphin_ctc_models(repo_id: str, decoding_method: str, num_active_paths
1029
  subfolder=".",
1030
  )
1031
 
 
 
 
 
 
 
 
 
1032
  recognizer = sherpa_onnx.OfflineRecognizer.from_dolphin_ctc(
1033
  tokens=tokens,
1034
  model=nn_model,
 
1008
  "csukuangfj/sherpa-onnx-dolphin-small-ctc-multi-lang-int8-2025-04-02",
1009
  "csukuangfj/sherpa-onnx-dolphin-base-ctc-multi-lang-2025-04-02",
1010
  "csukuangfj/sherpa-onnx-dolphin-small-ctc-multi-lang-2025-04-02",
1011
+ "csukuangfj/sherpa-onnx-nemo-canary-180m-flash-en-es-de-fr-int8",
1012
  ], repo_id
1013
 
1014
  if repo_id in [
 
1030
  subfolder=".",
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,