csukuangfj commited on
Commit
9fa23bb
1 Parent(s): 4abdf19

add new CTC model

Browse files
Files changed (1) hide show
  1. model.py +6 -1
model.py CHANGED
@@ -187,6 +187,7 @@ def _get_english_model(
187
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11", # noqa
188
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14", # noqa
189
  "videodanchik/icefall-asr-tedlium3-conformer-ctc2",
 
190
  ], repo_id
191
 
192
  filename = "cpu_jit.pt"
@@ -208,7 +209,10 @@ def _get_english_model(
208
  )
209
  subfolder = "data/lang_bpe_500"
210
 
211
- if repo_id == "videodanchik/icefall-asr-tedlium3-conformer-ctc2":
 
 
 
212
  subfolder = "data/lang_bpe"
213
 
214
  tokens = _get_token_filename(repo_id=repo_id, subfolder=subfolder)
@@ -559,6 +563,7 @@ english_models = {
559
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_english_model, # noqa
560
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13": _get_english_model, # noqa
561
  "videodanchik/icefall-asr-tedlium3-conformer-ctc2": _get_english_model,
 
562
  "csukuangfj/wenet-english-model": _get_wenet_model,
563
  }
564
 
 
187
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11", # noqa
188
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14", # noqa
189
  "videodanchik/icefall-asr-tedlium3-conformer-ctc2",
190
+ "pkufool/icefall_asr_librispeech_conformer_ctc",
191
  ], repo_id
192
 
193
  filename = "cpu_jit.pt"
 
209
  )
210
  subfolder = "data/lang_bpe_500"
211
 
212
+ if repo_id in (
213
+ "videodanchik/icefall-asr-tedlium3-conformer-ctc2",
214
+ "pkufool/icefall_asr_librispeech_conformer_ctc",
215
+ ):
216
  subfolder = "data/lang_bpe"
217
 
218
  tokens = _get_token_filename(repo_id=repo_id, subfolder=subfolder)
 
563
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_english_model, # noqa
564
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13": _get_english_model, # noqa
565
  "videodanchik/icefall-asr-tedlium3-conformer-ctc2": _get_english_model,
566
+ "pkufool/icefall_asr_librispeech_conformer_ctc": _get_english_model,
567
  "csukuangfj/wenet-english-model": _get_wenet_model,
568
  }
569