Spaces:
Build error
Build error
Commit
·
451aa75
1
Parent(s):
6c15050
update model from ricky
Browse files
model.py
CHANGED
|
@@ -124,6 +124,7 @@ def _get_gigaspeech_pre_trained_model(repo_id: str) -> OfflineAsr:
|
|
| 124 |
@lru_cache(maxsize=10)
|
| 125 |
def _get_librispeech_pre_trained_model(repo_id: str) -> OfflineAsr:
|
| 126 |
assert repo_id in [
|
|
|
|
| 127 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13", # noqa
|
| 128 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11", # noqa
|
| 129 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14", # noqa
|
|
@@ -136,6 +137,12 @@ def _get_librispeech_pre_trained_model(repo_id: str) -> OfflineAsr:
|
|
| 136 |
):
|
| 137 |
filename = "cpu_jit-torch-1.10.0.pt"
|
| 138 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 139 |
nn_model_filename = _get_nn_model_filename(
|
| 140 |
repo_id=repo_id,
|
| 141 |
filename=filename,
|
|
@@ -266,6 +273,7 @@ chinese_models = {
|
|
| 266 |
|
| 267 |
english_models = {
|
| 268 |
"wgb14/icefall-asr-gigaspeech-pruned-transducer-stateless2": _get_gigaspeech_pre_trained_model, # noqa
|
|
|
|
| 269 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14": _get_librispeech_pre_trained_model, # noqa
|
| 270 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_librispeech_pre_trained_model, # noqa
|
| 271 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13": _get_librispeech_pre_trained_model, # noqa
|
|
|
|
| 124 |
@lru_cache(maxsize=10)
|
| 125 |
def _get_librispeech_pre_trained_model(repo_id: str) -> OfflineAsr:
|
| 126 |
assert repo_id in [
|
| 127 |
+
"WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02", # noqa
|
| 128 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13", # noqa
|
| 129 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11", # noqa
|
| 130 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14", # noqa
|
|
|
|
| 137 |
):
|
| 138 |
filename = "cpu_jit-torch-1.10.0.pt"
|
| 139 |
|
| 140 |
+
if (
|
| 141 |
+
repo_id
|
| 142 |
+
== "WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02"
|
| 143 |
+
):
|
| 144 |
+
filename = "cpu_jit-torch-1.10.pt"
|
| 145 |
+
|
| 146 |
nn_model_filename = _get_nn_model_filename(
|
| 147 |
repo_id=repo_id,
|
| 148 |
filename=filename,
|
|
|
|
| 273 |
|
| 274 |
english_models = {
|
| 275 |
"wgb14/icefall-asr-gigaspeech-pruned-transducer-stateless2": _get_gigaspeech_pre_trained_model, # noqa
|
| 276 |
+
"WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02": _get_librispeech_pre_trained_model, # noqa
|
| 277 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14": _get_librispeech_pre_trained_model, # noqa
|
| 278 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_librispeech_pre_trained_model, # noqa
|
| 279 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13": _get_librispeech_pre_trained_model, # noqa
|