csukuangfj commited on
Commit
6281348
·
1 Parent(s): 480f8c2

add medium.en

Browse files
Files changed (1) hide show
  1. model.py +2 -0
model.py CHANGED
@@ -96,6 +96,7 @@ def get_pretrained_model(name: str) -> sherpa_onnx.OfflineRecognizer:
96
  "tiny.en",
97
  "base.en",
98
  "small.en",
 
99
  "tiny",
100
  "base",
101
  "small",
@@ -128,6 +129,7 @@ whisper_models = {
128
  "tiny.en": get_pretrained_model,
129
  "base.en": get_pretrained_model,
130
  "small.en": get_pretrained_model,
 
131
  "tiny": get_pretrained_model,
132
  "base": get_pretrained_model,
133
  "small": get_pretrained_model,
 
96
  "tiny.en",
97
  "base.en",
98
  "small.en",
99
+ "medium.en",
100
  "tiny",
101
  "base",
102
  "small",
 
129
  "tiny.en": get_pretrained_model,
130
  "base.en": get_pretrained_model,
131
  "small.en": get_pretrained_model,
132
+ "medium.en": get_pretrained_model,
133
  "tiny": get_pretrained_model,
134
  "base": get_pretrained_model,
135
  "small": get_pretrained_model,