Update model.py
Browse files
model.py
CHANGED
|
@@ -523,22 +523,22 @@ def _get_revolab_pretrained_model(
|
|
| 523 |
|
| 524 |
encoder_model = _get_nn_model_filename(
|
| 525 |
repo_id=real_repo,
|
| 526 |
-
filename="encoder-epoch-
|
| 527 |
-
subfolder="zipformer-large-finetune-
|
| 528 |
)
|
| 529 |
decoder_model = _get_nn_model_filename(
|
| 530 |
repo_id=real_repo,
|
| 531 |
-
filename="decoder-epoch-
|
| 532 |
-
subfolder="zipformer-large-finetune-
|
| 533 |
)
|
| 534 |
|
| 535 |
joiner_model = _get_nn_model_filename(
|
| 536 |
repo_id=real_repo,
|
| 537 |
-
filename="joiner-epoch-
|
| 538 |
-
subfolder="zipformer-large-finetune-
|
| 539 |
)
|
| 540 |
|
| 541 |
-
tokens = _get_token_filename(repo_id=real_repo, subfolder="zipformer-large-
|
| 542 |
|
| 543 |
recognizer = sherpa_onnx.OfflineRecognizer.from_transducer(
|
| 544 |
tokens=tokens,
|
|
|
|
| 523 |
|
| 524 |
encoder_model = _get_nn_model_filename(
|
| 525 |
repo_id=real_repo,
|
| 526 |
+
filename="encoder-epoch-6-avg-1.onnx",
|
| 527 |
+
subfolder="zipformer-large-finetune-CTC/export",
|
| 528 |
)
|
| 529 |
decoder_model = _get_nn_model_filename(
|
| 530 |
repo_id=real_repo,
|
| 531 |
+
filename="decoder-epoch-6-avg-1.onnx",
|
| 532 |
+
subfolder="zipformer-large-finetune-CTC/export",
|
| 533 |
)
|
| 534 |
|
| 535 |
joiner_model = _get_nn_model_filename(
|
| 536 |
repo_id=real_repo,
|
| 537 |
+
filename="joiner-epoch-6-avg-1.onnx",
|
| 538 |
+
subfolder="zipformer-large-finetune-CTC/export",
|
| 539 |
)
|
| 540 |
|
| 541 |
+
tokens = _get_token_filename(repo_id=real_repo, subfolder="zipformer-large-finetune-CTC/exp/lang_bpe_500")
|
| 542 |
|
| 543 |
recognizer = sherpa_onnx.OfflineRecognizer.from_transducer(
|
| 544 |
tokens=tokens,
|