add wenet models

model.py

@@ -327,6 +327,46 @@ def _get_alimeeting_pre_trained_model(
     return recognizer
 
 
+@lru_cache(maxsize=10)
+def _get_wenet_model(
+    repo_id: str,
+    decoding_method: str,
+    num_active_paths: int,
+):
+    assert repo_id in [
+        "csukuangfj/wenet-chinese-model",
+    ], repo_id
+
+    nn_model = _get_nn_model_filename(
+        repo_id=repo_id,
+        filename="final.zip",
+        subfolder="./",
+    )
+    tokens = _get_token_filename(
+        repo_id=repo_id,
+        filename="units.txt",
+        subfolder="./",
+    )
+
+    feat_config = sherpa.FeatureConfig(normalize_samples=False)
+    feat_config.fbank_opts.frame_opts.samp_freq = sample_rate
+    feat_config.fbank_opts.mel_opts.num_bins = 80
+    feat_config.fbank_opts.frame_opts.dither = 0
+
+    config = sherpa.OfflineRecognizerConfig(
+        nn_model=nn_model,
+        tokens=tokens,
+        use_gpu=False,
+        feat_config=feat_config,
+        decoding_method=decoding_method,
+        num_active_paths=num_active_paths,
+    )
+
+    recognizer = sherpa.OfflineRecognizer(config)
+
+    return recognizer
+
+
 @lru_cache(maxsize=10)
 def _get_aidatatang_200zh_pretrained_mode(
     repo_id: str,
@@ -448,6 +488,7 @@ chinese_models = {
     "yuekai/icefall-asr-aishell2-pruned-transducer-stateless5-B-2022-07-12": _get_aishell2_pretrained_model,  # noqa
     "luomingshuang/icefall_asr_aidatatang-200zh_pruned_transducer_stateless2": _get_aidatatang_200zh_pretrained_mode,  # noqa
     "luomingshuang/icefall_asr_alimeeting_pruned_transducer_stateless2": _get_alimeeting_pre_trained_model,  # noqa
+    "csukuangfj/wenet-chinese-model": _get_wenet_model,
 }
 
 english_models = {
@@ -456,6 +497,7 @@ english_models = {
     "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14": _get_librispeech_pre_trained_model,  # noqa
     "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_librispeech_pre_trained_model,  # noqa
     "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13": _get_librispeech_pre_trained_model,  # noqa
+    "csukuangfj/wenet-english-model": _get_wenet_model,
 }
 
 chinese_english_mixed_models = {