csukuangfj commited on
Commit
a6f60a8
·
1 Parent(s): 5b99533

Add a new model

Browse files
Files changed (1) hide show
  1. model.py +11 -0
model.py CHANGED
@@ -182,6 +182,10 @@ def get_pretrained_model(
182
  return chinese_models[repo_id](
183
  repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
184
  )
 
 
 
 
185
  elif repo_id in chinese_dialect_models:
186
  return chinese_dialect_models[repo_id](
187
  repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
@@ -2008,6 +2012,7 @@ def _get_paraformer_pre_trained_model(
2008
  "csukuangfj/sherpa-onnx-paraformer-zh-small-2024-03-09",
2009
  "csukuangfj/sherpa-onnx-paraformer-trilingual-zh-cantonese-en",
2010
  "csukuangfj/sherpa-onnx-paraformer-en-2024-03-09",
 
2011
  ], repo_id
2012
 
2013
  nn_model = _get_nn_model_filename(
@@ -2147,6 +2152,10 @@ chinese_dialect_models = {
2147
  "csukuangfj/sherpa-onnx-telespeech-ctc-int8-zh-2024-06-04": _get_chinese_dialect_models,
2148
  }
2149
 
 
 
 
 
2150
  chinese_models = {
2151
  "csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-int8-2025-06-30": _get_streaming_zipformer_ctc_pre_trained_model,
2152
  "csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-2025-06-30": _get_streaming_zipformer_ctc_pre_trained_model,
@@ -2319,6 +2328,7 @@ portuguese_brazlian_models = {
2319
  all_models = {
2320
  **twenty_five_languages_models,
2321
  **multi_lingual_models,
 
2322
  **chinese_models,
2323
  **english_models,
2324
  **chinese_english_mixed_models,
@@ -2344,6 +2354,7 @@ language_to_models = {
2344
  "超多种中文方言": list(chinese_dialect_models.keys()),
2345
  "Chinese": list(chinese_models.keys()),
2346
  "English": list(english_models.keys()),
 
2347
  "Chinese+English": list(chinese_english_mixed_models.keys()),
2348
  "Chinese+English+Cantonese": list(chinese_cantonese_english_models.keys()),
2349
  "Chinese+English+Cantonese+Japanese+Korean": list(
 
182
  return chinese_models[repo_id](
183
  repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
184
  )
185
+ elif repo_id in sichuan_models:
186
+ return sichuan_models[repo_id](
187
+ repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
188
+ )
189
  elif repo_id in chinese_dialect_models:
190
  return chinese_dialect_models[repo_id](
191
  repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
 
2012
  "csukuangfj/sherpa-onnx-paraformer-zh-small-2024-03-09",
2013
  "csukuangfj/sherpa-onnx-paraformer-trilingual-zh-cantonese-en",
2014
  "csukuangfj/sherpa-onnx-paraformer-en-2024-03-09",
2015
+ "csukuangfj/sherpa-onnx-paraformer-zh-int8-2025-10-07",
2016
  ], repo_id
2017
 
2018
  nn_model = _get_nn_model_filename(
 
2152
  "csukuangfj/sherpa-onnx-telespeech-ctc-int8-zh-2024-06-04": _get_chinese_dialect_models,
2153
  }
2154
 
2155
+ sichuan_models = {
2156
+ "csukuangfj/sherpa-onnx-paraformer-zh-int8-2025-10-07": _get_paraformer_pre_trained_model,
2157
+ }
2158
+
2159
  chinese_models = {
2160
  "csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-int8-2025-06-30": _get_streaming_zipformer_ctc_pre_trained_model,
2161
  "csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-2025-06-30": _get_streaming_zipformer_ctc_pre_trained_model,
 
2328
  all_models = {
2329
  **twenty_five_languages_models,
2330
  **multi_lingual_models,
2331
+ **sichuan_models,
2332
  **chinese_models,
2333
  **english_models,
2334
  **chinese_english_mixed_models,
 
2354
  "超多种中文方言": list(chinese_dialect_models.keys()),
2355
  "Chinese": list(chinese_models.keys()),
2356
  "English": list(english_models.keys()),
2357
+ "四川话": list(sichuan_models.keys()),
2358
  "Chinese+English": list(chinese_english_mixed_models.keys()),
2359
  "Chinese+English+Cantonese": list(chinese_cantonese_english_models.keys()),
2360
  "Chinese+English+Cantonese+Japanese+Korean": list(