add aidatatang_200zh

examples.py

@@ -130,6 +130,29 @@ examples = [
         4,
         "./test_wavs/aishell2/ID0012W0215.wav",
     ],
+    # aishell2-B
+    # https://huggingface.co/luomingshuang/icefall_asr_aidatatang-200zh_pruned_transducer_stateless2/tree/main/test_wavs
+    [
+        "Chinese",
+        "luomingshuang/icefall_asr_aidatatang-200zh_pruned_transducer_stateless2",
+        "greedy_search",
+        4,
+        "./test_wavs/aidatatang_200zh/T0055G0036S0002.wav",
+    ],
+    [
+        "Chinese",
+        "luomingshuang/icefall_asr_aidatatang-200zh_pruned_transducer_stateless2",
+        "greedy_search",
+        4,
+        "./test_wavs/aidatatang_200zh/T0055G0036S0003.wav",
+    ],
+    [
+        "Chinese",
+        "luomingshuang/icefall_asr_aidatatang-200zh_pruned_transducer_stateless2",
+        "greedy_search",
+        4,
+        "./test_wavs/aidatatang_200zh/T0055G0036S0004.wav",
+    ],
     # tal_csasr
     # https://huggingface.co/luomingshuang/icefall_asr_tal-csasr_pruned_transducer_stateless5/tree/main/test_wavs
     [

model.py

@@ -203,11 +203,33 @@ def _get_alimeeting_pre_trained_model(repo_id: str):
     )
 
 
+@lru_cache(maxsize=10)
+def _get_aidatatang_200zh_pretrained_mode(repo_id: str):
+    assert repo_id in [
+        "luomingshuang/icefall_asr_aidatatang-200zh_pruned_transducer_stateless2",
+    ], repo_id
+
+    nn_model_filename = _get_nn_model_filename(
+        repo_id=repo_id,
+        filename="cpu_jit_torch_1.7.1.pt",
+    )
+    token_filename = _get_token_filename(repo_id=repo_id)
+
+    return OfflineAsr(
+        nn_model_filename=nn_model_filename,
+        bpe_model_filename=None,
+        token_filename=token_filename,
+        sample_rate=sample_rate,
+        device="cpu",
+    )
+
+
 chinese_models = {
     "luomingshuang/icefall_asr_wenetspeech_pruned_transducer_stateless2": _get_wenetspeech_pre_trained_model,  # noqa
     "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
     "luomingshuang/icefall_asr_alimeeting_pruned_transducer_stateless2": _get_alimeeting_pre_trained_model,  # noqa
+    "luomingshuang/icefall_asr_aidatatang-200zh_pruned_transducer_stateless2": _get_aidatatang_200zh_pretrained_mode,  # noqa
 }
 
 english_models = {

test_wavs/aidatatang_200zh/README.md

@@ -0,0 +1,2 @@
+Files are downloaded from
+https://huggingface.co/luomingshuang/icefall_asr_aidatatang-200zh_pruned_transducer_stateless2/tree/main/test_wavs