aisyahrevolab commited on
Commit
081f8ed
·
verified ·
1 Parent(s): e235451

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +36 -0
model.py CHANGED
@@ -588,6 +588,42 @@ def _get_revolab_pretrained_model(
588
 
589
  return recognizer
590
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
591
  @lru_cache(maxsize=10)
592
  def _get_zrjin_cantonese_pre_trained_model(
593
  repo_id: str, decoding_method: str, num_active_paths: int
 
588
 
589
  return recognizer
590
 
591
+ elif repo_id == "Revolab/zipformer-scaled-up":
592
+
593
+ real_repo = 'Revolab/malaysian-pruned_transducer_stateless7'
594
+
595
+ encoder_model = _get_nn_model_filename(
596
+ repo_id=real_repo,
597
+ filename="encoder-epoch-13-avg-5.onnx",
598
+ subfolder="zipformer-scale-up-pretrained/export",
599
+ )
600
+ decoder_model = _get_nn_model_filename(
601
+ repo_id=real_repo,
602
+ filename="encoder-epoch-13-avg-5.onnx",
603
+ subfolder="zipformer-scale-up-pretrained/export",
604
+ )
605
+
606
+ joiner_model = _get_nn_model_filename(
607
+ repo_id=real_repo,
608
+ filename="joiner-epoch-13-avg-5.onnx",
609
+ subfolder="zipformer-scale-up-pretrained/export",
610
+ )
611
+
612
+ tokens = _get_token_filename(repo_id=real_repo, subfolder="zipformer-scale-up-pretrained/export/lang_bpe_500")
613
+
614
+ recognizer = sherpa_onnx.OfflineRecognizer.from_transducer(
615
+ tokens=tokens,
616
+ encoder=encoder_model,
617
+ decoder=decoder_model,
618
+ joiner=joiner_model,
619
+ num_threads=2,
620
+ sample_rate=16000,
621
+ feature_dim=80,
622
+ decoding_method=decoding_method,
623
+ )
624
+
625
+ return recognizer
626
+
627
  @lru_cache(maxsize=10)
628
  def _get_zrjin_cantonese_pre_trained_model(
629
  repo_id: str, decoding_method: str, num_active_paths: int