csukuangfj commited on
Commit
46f8085
·
1 Parent(s): 7fd804c

add more models

Browse files
Files changed (1) hide show
  1. model.py +18 -1
model.py CHANGED
@@ -214,6 +214,10 @@ def get_pretrained_model(
214
  return french_models[repo_id](
215
  repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
216
  )
 
 
 
 
217
  elif repo_id in japanese_models:
218
  return japanese_models[repo_id](
219
  repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
@@ -1587,9 +1591,15 @@ def _get_streaming_zipformer_pre_trained_model(
1587
  "csukuangfj/sherpa-onnx-streaming-zipformer-bilingual-zh-en-2023-02-20",
1588
  "k2-fsa/sherpa-onnx-streaming-zipformer-korean-2024-06-16",
1589
  "csukuangfj/sherpa-onnx-streaming-zipformer-en-kroko-2025-08-06",
 
 
1590
  ], repo_id
1591
 
1592
- if repo_id == "csukuangfj/sherpa-onnx-streaming-zipformer-en-kroko-2025-08-06":
 
 
 
 
1593
  encoder_model = _get_nn_model_filename(
1594
  repo_id=repo_id,
1595
  filename="encoder.onnx",
@@ -2112,6 +2122,11 @@ german_models = {
2112
 
2113
  french_models = {
2114
  "shaojieli/sherpa-onnx-streaming-zipformer-fr-2023-04-14": _get_french_pre_trained_model,
 
 
 
 
 
2115
  }
2116
 
2117
  japanese_models = {
@@ -2177,6 +2192,7 @@ all_models = {
2177
  **arabic_models,
2178
  **german_models,
2179
  **french_models,
 
2180
  **russian_models,
2181
  **korean_models,
2182
  **thai_models,
@@ -2202,6 +2218,7 @@ language_to_models = {
2202
  "Korean": list(korean_models.keys()),
2203
  "Portuguese (Brazil)": list(portuguese_brazlian_models.keys()),
2204
  "Russian": list(russian_models.keys()),
 
2205
  "Thai": list(thai_models.keys()),
2206
  "Tibetan": list(tibetan_models.keys()),
2207
  "Vietnamese": list(vietnamese_models.keys()),
 
214
  return french_models[repo_id](
215
  repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
216
  )
217
+ elif repo_id in spanish_models:
218
+ return spanish_models[repo_id](
219
+ repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
220
+ )
221
  elif repo_id in japanese_models:
222
  return japanese_models[repo_id](
223
  repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
 
1591
  "csukuangfj/sherpa-onnx-streaming-zipformer-bilingual-zh-en-2023-02-20",
1592
  "k2-fsa/sherpa-onnx-streaming-zipformer-korean-2024-06-16",
1593
  "csukuangfj/sherpa-onnx-streaming-zipformer-en-kroko-2025-08-06",
1594
+ "csukuangfj/sherpa-onnx-streaming-zipformer-es-kroko-2025-08-06",
1595
+ "csukuangfj/sherpa-onnx-streaming-zipformer-fr-kroko-2025-08-06",
1596
  ], repo_id
1597
 
1598
+ if repo_id in [
1599
+ "csukuangfj/sherpa-onnx-streaming-zipformer-en-kroko-2025-08-06",
1600
+ "csukuangfj/sherpa-onnx-streaming-zipformer-es-kroko-2025-08-06",
1601
+ "csukuangfj/sherpa-onnx-streaming-zipformer-fr-kroko-2025-08-06",
1602
+ ]:
1603
  encoder_model = _get_nn_model_filename(
1604
  repo_id=repo_id,
1605
  filename="encoder.onnx",
 
2122
 
2123
  french_models = {
2124
  "shaojieli/sherpa-onnx-streaming-zipformer-fr-2023-04-14": _get_french_pre_trained_model,
2125
+ "csukuangfj/sherpa-onnx-streaming-zipformer-fr-kroko-2025-08-06": _get_streaming_zipformer_pre_trained_model,
2126
+ }
2127
+
2128
+ spanish_models = {
2129
+ "csukuangfj/sherpa-onnx-streaming-zipformer-es-kroko-2025-08-06": _get_streaming_zipformer_pre_trained_model,
2130
  }
2131
 
2132
  japanese_models = {
 
2192
  **arabic_models,
2193
  **german_models,
2194
  **french_models,
2195
+ **spanish_models,
2196
  **russian_models,
2197
  **korean_models,
2198
  **thai_models,
 
2218
  "Korean": list(korean_models.keys()),
2219
  "Portuguese (Brazil)": list(portuguese_brazlian_models.keys()),
2220
  "Russian": list(russian_models.keys()),
2221
+ "Spanish": list(spanish_models.keys()),
2222
  "Thai": list(thai_models.keys()),
2223
  "Tibetan": list(tibetan_models.keys()),
2224
  "Vietnamese": list(vietnamese_models.keys()),