update models

model.py

@@ -794,7 +794,7 @@ def _get_french_pre_trained_model(
     repo_id: str,
     decoding_method: str,
     num_active_paths: int,
-) -> sherpa_onnx.OfflineRecognizer:
+) -> sherpa_onnx.OnlineRecognizer:
     assert repo_id in [
         "shaojieli/sherpa-onnx-streaming-zipformer-fr-2023-04-14",
     ], repo_id
@@ -834,6 +834,51 @@ def _get_french_pre_trained_model(
     return recognizer
 
 
+@lru_cache(maxsize=10)
+def _get_streaming_zipformer_pre_trained_model(
+    repo_id: str,
+    decoding_method: str,
+    num_active_paths: int,
+) -> sherpa_onnx.OnlineRecognizer:
+    assert repo_id in [
+        "csukuangfj/sherpa-onnx-streaming-zipformer-bilingual-zh-en-2023-02-20",
+    ], repo_id
+
+    encoder_model = _get_nn_model_filename(
+        repo_id=repo_id,
+        filename="encoder-epoch-99-avg-1.onnx",
+        subfolder=".",
+    )
+
+    decoder_model = _get_nn_model_filename(
+        repo_id=repo_id,
+        filename="decoder-epoch-99-avg-1.onnx",
+        subfolder=".",
+    )
+
+    joiner_model = _get_nn_model_filename(
+        repo_id=repo_id,
+        filename="joiner-epoch-99-avg-1.onnx",
+        subfolder=".",
+    )
+
+    tokens = _get_token_filename(repo_id=repo_id, subfolder=".")
+
+    recognizer = sherpa_onnx.OnlineRecognizer.from_transducer(
+        tokens=tokens,
+        encoder=encoder_model,
+        decoder=decoder_model,
+        joiner=joiner_model,
+        num_threads=2,
+        sample_rate=16000,
+        feature_dim=80,
+        decoding_method=decoding_method,
+        max_active_paths=num_active_paths,
+    )
+
+    return recognizer
+
+
 @lru_cache(maxsize=10)
 def _get_japanese_pre_trained_model(
     repo_id: str,
@@ -1007,6 +1052,8 @@ def _get_paraformer_pre_trained_model(
 ) -> sherpa_onnx.OfflineRecognizer:
     assert repo_id in [
         "csukuangfj/sherpa-onnx-paraformer-zh-2023-03-28",
+        "csukuangfj/sherpa-onnx-paraformer-zh-2024-03-09",
+        "csukuangfj/sherpa-onnx-paraformer-zh-small-2024-03-09",
         "csukuangfj/sherpa-onnx-paraformer-trilingual-zh-cantonese-en",
     ], repo_id
 
@@ -1129,10 +1176,9 @@ def _get_multi_zh_hans_pre_trained_model(
 
 
 chinese_models = {
-    "csukuangfj/sherpa-onnx-paraformer-zh-2023-03-28": _get_paraformer_pre_trained_model,
-    "csukuangfj/sherpa-onnx-paraformer-zh-small-2024-03-09": _get_paraformer_pre_trained_model,
     "csukuangfj/sherpa-onnx-paraformer-zh-2024-03-09": _get_paraformer_pre_trained_model,
     "luomingshuang/icefall_asr_wenetspeech_pruned_transducer_stateless2": _get_wenetspeech_pre_trained_model,  # noqa
+    "csukuangfj/sherpa-onnx-paraformer-zh-small-2024-03-09": _get_paraformer_pre_trained_model,
     "zrjin/sherpa-onnx-zipformer-multi-zh-hans-2023-9-2": _get_multi_zh_hans_pre_trained_model,  # noqa
     "zrjin/icefall-asr-aishell-zipformer-large-2023-10-24": _get_aishell_pre_trained_model,  # noqa
     "zrjin/icefall-asr-aishell-zipformer-small-2023-10-24": _get_aishell_pre_trained_model,  # noqa
@@ -1171,6 +1217,8 @@ english_models = {
 }
 
 chinese_english_mixed_models = {
+    "csukuangfj/sherpa-onnx-streaming-zipformer-bilingual-zh-en-2023-02-20": _get_streaming_zipformer_pre_trained_model,
+    "csukuangfj/sherpa-onnx-paraformer-zh-2023-03-28": _get_paraformer_pre_trained_model,
     "ptrnull/icefall-asr-conv-emformer-transducer-stateless2-zh": _get_chinese_english_mixed_model,
     "luomingshuang/icefall_asr_tal-csasr_pruned_transducer_stateless5": _get_chinese_english_mixed_model,  # noqa
 }
@@ -1203,8 +1251,8 @@ russian_models = {
 }
 
 chinese_cantonese_english_models = {
-    "csukuangfj/sherpa-onnx-streaming-paraformer-trilingual-zh-cantonese-en": _get_streaming_paraformer_zh_yue_en_pre_trained_model,
     "csukuangfj/sherpa-onnx-paraformer-trilingual-zh-cantonese-en": _get_paraformer_pre_trained_model,
+    "csukuangfj/sherpa-onnx-streaming-paraformer-trilingual-zh-cantonese-en": _get_streaming_paraformer_zh_yue_en_pre_trained_model,
 }