csukuangfj
commited on
Commit
•
cce0d1e
1
Parent(s):
70e1861
add medium-aishell
Browse files
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 |
}
|