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 |
}
|