Spaces:
Build error
Build error
Commit
·
0ae65b0
1
Parent(s):
9edf22d
add a new multidataset zipformer model
Browse files
model.py
CHANGED
|
@@ -283,6 +283,7 @@ def _get_english_model(
|
|
| 283 |
) -> sherpa.OfflineRecognizer:
|
| 284 |
assert repo_id in [
|
| 285 |
"WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02", # noqa
|
|
|
|
| 286 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13", # noqa
|
| 287 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11", # noqa
|
| 288 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14", # noqa
|
|
@@ -304,6 +305,12 @@ def _get_english_model(
|
|
| 304 |
):
|
| 305 |
filename = "cpu_jit-torch-1.10.pt"
|
| 306 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 307 |
nn_model = _get_nn_model_filename(
|
| 308 |
repo_id=repo_id,
|
| 309 |
filename=filename,
|
|
@@ -742,6 +749,7 @@ chinese_models = {
|
|
| 742 |
|
| 743 |
english_models = {
|
| 744 |
"wgb14/icefall-asr-gigaspeech-pruned-transducer-stateless2": _get_gigaspeech_pre_trained_model, # noqa
|
|
|
|
| 745 |
"WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02": _get_english_model, # noqa
|
| 746 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14": _get_english_model, # noqa
|
| 747 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_english_model, # noqa
|
|
|
|
| 283 |
) -> sherpa.OfflineRecognizer:
|
| 284 |
assert repo_id in [
|
| 285 |
"WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02", # noqa
|
| 286 |
+
"yfyeung/icefall-asr-multidataset-pruned_transducer_stateless7-2023-05-04", # noqa
|
| 287 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13", # noqa
|
| 288 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11", # noqa
|
| 289 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14", # noqa
|
|
|
|
| 305 |
):
|
| 306 |
filename = "cpu_jit-torch-1.10.pt"
|
| 307 |
|
| 308 |
+
if (
|
| 309 |
+
repo_id
|
| 310 |
+
== "yfyeung/icefall-asr-multidataset-pruned_transducer_stateless7-2023-05-04"
|
| 311 |
+
):
|
| 312 |
+
filename = "cpu_jit-epoch-30-avg-4.pt"
|
| 313 |
+
|
| 314 |
nn_model = _get_nn_model_filename(
|
| 315 |
repo_id=repo_id,
|
| 316 |
filename=filename,
|
|
|
|
| 749 |
|
| 750 |
english_models = {
|
| 751 |
"wgb14/icefall-asr-gigaspeech-pruned-transducer-stateless2": _get_gigaspeech_pre_trained_model, # noqa
|
| 752 |
+
"yfyeung/icefall-asr-multidataset-pruned_transducer_stateless7-2023-05-04": _get_english_model, # noqa
|
| 753 |
"WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02": _get_english_model, # noqa
|
| 754 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14": _get_english_model, # noqa
|
| 755 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_english_model, # noqa
|