Spaces:
Build error
Build error
Commit
·
9b54310
1
Parent(s):
3069780
update english + chinese model
Browse files
model.py
CHANGED
|
@@ -262,20 +262,28 @@ def _get_wenetspeech_pre_trained_model(
|
|
| 262 |
|
| 263 |
|
| 264 |
@lru_cache(maxsize=10)
|
| 265 |
-
def
|
| 266 |
repo_id: str,
|
| 267 |
decoding_method: str,
|
| 268 |
num_active_paths: int,
|
| 269 |
):
|
| 270 |
assert repo_id in [
|
| 271 |
"luomingshuang/icefall_asr_tal-csasr_pruned_transducer_stateless5",
|
|
|
|
| 272 |
], repo_id
|
| 273 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 274 |
nn_model = _get_nn_model_filename(
|
| 275 |
repo_id=repo_id,
|
| 276 |
-
filename=
|
| 277 |
)
|
| 278 |
-
tokens = _get_token_filename(repo_id=repo_id)
|
| 279 |
|
| 280 |
feat_config = sherpa.FeatureConfig()
|
| 281 |
feat_config.fbank_opts.frame_opts.samp_freq = sample_rate
|
|
@@ -541,7 +549,8 @@ english_models = {
|
|
| 541 |
}
|
| 542 |
|
| 543 |
chinese_english_mixed_models = {
|
| 544 |
-
"luomingshuang/icefall_asr_tal-csasr_pruned_transducer_stateless5":
|
|
|
|
| 545 |
}
|
| 546 |
|
| 547 |
tibetan_models = {
|
|
|
|
| 262 |
|
| 263 |
|
| 264 |
@lru_cache(maxsize=10)
|
| 265 |
+
def _get_chinese_english_mixed_model(
|
| 266 |
repo_id: str,
|
| 267 |
decoding_method: str,
|
| 268 |
num_active_paths: int,
|
| 269 |
):
|
| 270 |
assert repo_id in [
|
| 271 |
"luomingshuang/icefall_asr_tal-csasr_pruned_transducer_stateless5",
|
| 272 |
+
"ptrnull/icefall-asr-conv-emformer-transducer-stateless2-zh",
|
| 273 |
], repo_id
|
| 274 |
|
| 275 |
+
if repo_id == "luomingshuang/icefall_asr_tal-csasr_pruned_transducer_stateless5":
|
| 276 |
+
filename = "cpu_jit.pt"
|
| 277 |
+
subfolder = "data/lang_char"
|
| 278 |
+
elif repo_id == "ptrnull/icefall-asr-conv-emformer-transducer-stateless2-zh":
|
| 279 |
+
filename = "cpu_jit-epoch-11-avg-1.pt"
|
| 280 |
+
subfolder = "data/lang_char_bpe"
|
| 281 |
+
|
| 282 |
nn_model = _get_nn_model_filename(
|
| 283 |
repo_id=repo_id,
|
| 284 |
+
filename=filename,
|
| 285 |
)
|
| 286 |
+
tokens = _get_token_filename(repo_id=repo_id, subfolder=subfolder)
|
| 287 |
|
| 288 |
feat_config = sherpa.FeatureConfig()
|
| 289 |
feat_config.fbank_opts.frame_opts.samp_freq = sample_rate
|
|
|
|
| 549 |
}
|
| 550 |
|
| 551 |
chinese_english_mixed_models = {
|
| 552 |
+
"luomingshuang/icefall_asr_tal-csasr_pruned_transducer_stateless5": _get_chinese_english_mixed_model, # noqa
|
| 553 |
+
"ptrnull/icefall-asr-conv-emformer-transducer-stateless2-zh": _get_chinese_english_mixed_model,
|
| 554 |
}
|
| 555 |
|
| 556 |
tibetan_models = {
|