csukuangfj commited on
Commit
cce0d1e
1 Parent(s): 70e1861

add medium-aishell

Browse files
Files changed (1) hide show
  1. model.py +2 -0
model.py CHANGED
@@ -101,6 +101,7 @@ def get_pretrained_model(name: str) -> sherpa_onnx.OfflineRecognizer:
101
  "base",
102
  "small",
103
  "medium",
 
104
  ), name
105
  full_repo_id = "csukuangfj/sherpa-onnx-whisper-" + name
106
  encoder = _get_nn_model_filename(
@@ -136,4 +137,5 @@ whisper_models = {
136
  "small": get_pretrained_model,
137
  "distil-small.en": get_pretrained_model,
138
  "medium": get_pretrained_model,
 
139
  }
 
101
  "base",
102
  "small",
103
  "medium",
104
+ "medium-aishell",
105
  ), name
106
  full_repo_id = "csukuangfj/sherpa-onnx-whisper-" + name
107
  encoder = _get_nn_model_filename(
 
137
  "small": get_pretrained_model,
138
  "distil-small.en": get_pretrained_model,
139
  "medium": get_pretrained_model,
140
+ "medium-aishell": get_pretrained_model,
141
  }