aisyahrevolab commited on
Commit
22a9cce
·
verified ·
1 Parent(s): bc6f9a5

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +7 -7
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-17-avg-1.onnx",
527
- subfolder="zipformer-large-finetune-SFO/export",
528
  )
529
  decoder_model = _get_nn_model_filename(
530
  repo_id=real_repo,
531
- filename="decoder-epoch-17-avg-1.onnx",
532
- subfolder="zipformer-large-finetune-SFO/export",
533
  )
534
 
535
  joiner_model = _get_nn_model_filename(
536
  repo_id=real_repo,
537
- filename="joiner-epoch-17-avg-1.onnx",
538
- subfolder="zipformer-large-finetune-SFO/export",
539
  )
540
 
541
- tokens = _get_token_filename(repo_id=real_repo, subfolder="zipformer-large-20k/exp/lang_bpe_500")
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,