Spaces:
Build error
Build error
Commit
·
9fa23bb
1
Parent(s):
4abdf19
add new CTC model
Browse files
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
|
|
|
|
|
|
|
|
|
|
| 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 |
|