update models

model.py

@@ -956,6 +956,61 @@ def _get_paraformer_zh_pre_trained_model(
     return recognizer
 
 
+def _get_aishell_pre_trained_model(
+    repo_id: str,
+    decoding_method: str,
+    num_active_paths: int,
+) -> sherpa_onnx.OfflineRecognizer:
+    assert repo_id in (
+        "zrjin/icefall-asr-aishell-zipformer-large-2023-10-24",
+        "zrjin/icefall-asr-aishell-zipformer-small-2023-10-24",
+        "zrjin/icefall-asr-aishell-zipformer-2023-10-24",
+    ), repo_id
+    if repo_id == "zrjin/icefall-asr-aishell-zipformer-large-2023-10-24":
+        epoch = 56
+        avg = 23
+    elif repo_id == "zrjin/icefall-asr-aishell-zipformer-small-2023-10-24":
+        epoch = 55
+        avg = 21
+    elif repo_id == "zrjin/icefall-asr-aishell-zipformer-2023-10-24":
+        epoch = 55
+        avg = 17
+
+    encoder_model = _get_nn_model_filename(
+        repo_id=repo_id,
+        filename=f"encoder-epoch-{epoch}-avg-{avg}.onnx",
+        subfolder="exp",
+    )
+
+    decoder_model = _get_nn_model_filename(
+        repo_id=repo_id,
+        filename=f"decoder-epoch-{epoch}-avg-{avg}.onnx",
+        subfolder="exp",
+    )
+
+    joiner_model = _get_nn_model_filename(
+        repo_id=repo_id,
+        filename=f"joiner-epoch-{epoch}-avg-{avg}.onnx",
+        subfolder="exp",
+    )
+
+    tokens = _get_token_filename(repo_id=repo_id, subfolder="data/lang_char")
+
+    recognizer = sherpa_onnx.OfflineRecognizer.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
+
+
 def _get_multi_zh_hans_pre_trained_model(
     repo_id: str,
     decoding_method: str,
@@ -1002,6 +1057,7 @@ chinese_models = {
     "csukuangfj/sherpa-onnx-paraformer-zh-2023-03-28": _get_paraformer_zh_pre_trained_model,
     "luomingshuang/icefall_asr_wenetspeech_pruned_transducer_stateless2": _get_wenetspeech_pre_trained_model,  # noqa
     "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
     "desh2608/icefall-asr-alimeeting-pruned-transducer-stateless7": _get_alimeeting_pre_trained_model,
     "yuekai/icefall-asr-aishell2-pruned-transducer-stateless5-A-2022-07-12": _get_aishell2_pretrained_model,  # noqa
     "yuekai/icefall-asr-aishell2-pruned-transducer-stateless5-B-2022-07-12": _get_aishell2_pretrained_model,  # noqa