add tedlium3 models

examples.py

@@ -79,6 +79,27 @@ examples = [
         4,
         "./test_wavs/alimeeting/R8009_M8020_N_SPK8026-8026-209.wav",
     ],
+    [
+        "English",
+        "videodanchik/icefall-asr-tedlium3-conformer-ctc2",
+        "greedy_search",
+        4,
+        "./test_wavs/tedlium3/DanBarber_2010-219.wav",
+    ],
+    [
+        "English",
+        "videodanchik/icefall-asr-tedlium3-conformer-ctc2",
+        "greedy_search",
+        4,
+        "./test_wavs/tedlium3/DanielKahneman_2010-157.wav",
+    ],
+    [
+        "English",
+        "videodanchik/icefall-asr-tedlium3-conformer-ctc2",
+        "greedy_search",
+        4,
+        "./test_wavs/tedlium3/RobertGupta_2010U-15.wav",
+    ],
     # librispeech
     # https://huggingface.co/csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless5-2022-05-13/tree/main/test_wavs
     [

model.py

@@ -176,7 +176,7 @@ def _get_gigaspeech_pre_trained_model(
 
 
 @lru_cache(maxsize=10)
-def _get_librispeech_pre_trained_model(
+def _get_english_model(
     repo_id: str,
     decoding_method: str,
     num_active_paths: int,
@@ -186,6 +186,7 @@ def _get_librispeech_pre_trained_model(
         "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13",  # noqa
         "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11",  # noqa
         "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14",  # noqa
+        "videodanchik/icefall-asr-tedlium3-conformer-ctc2",
     ], repo_id
 
     filename = "cpu_jit.pt"
@@ -205,7 +206,12 @@ def _get_librispeech_pre_trained_model(
         repo_id=repo_id,
         filename=filename,
     )
-    tokens = _get_token_filename(repo_id=repo_id, subfolder="data/lang_bpe_500")
+    subfolder = "data/lang_bpe_500"
+
+    if repo_id == "videodanchik/icefall-asr-tedlium3-conformer-ctc2":
+        subfolder = "data/lang_bpe"
+
+    tokens = _get_token_filename(repo_id=repo_id, subfolder=subfolder)
 
     feat_config = sherpa.FeatureConfig()
     feat_config.fbank_opts.frame_opts.samp_freq = sample_rate
@@ -548,10 +554,11 @@ chinese_models = {
 
 english_models = {
     "wgb14/icefall-asr-gigaspeech-pruned-transducer-stateless2": _get_gigaspeech_pre_trained_model,  # noqa
-    "WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02": _get_librispeech_pre_trained_model,  # noqa
-    "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
+    "WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02": _get_english_model,  # noqa
+    "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14": _get_english_model,  # noqa
+    "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_english_model,  # noqa
+    "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13": _get_english_model,  # noqa
+    "videodanchik/icefall-asr-tedlium3-conformer-ctc2": _get_english_model,
     "csukuangfj/wenet-english-model": _get_wenet_model,
 }
 

test_wavs/tedlium3/trans.txt

@@ -0,0 +1,3 @@
+DanBarber_2010-219 well the last year this property had six hundred thousand birds on it
+RobertGupta_2010U-15 and he was talking about invisible demons and smoke and how someone was poisoning him in his sleep and i was afraid not for myself but i was afraid that i was going to lose him that he was going to sink into one of his states
+DanielKahneman_2010-157 goes very different ways depending on how you think and whether you think of the remembering self or you think of the experiencing self this is going to influence policy i think in years to come in the united states efforts are being made