Spaces:
Running
Running
update
Browse files- main.py +1 -0
- toolbox/k2_sherpa/nn_models.py +32 -0
main.py
CHANGED
@@ -248,6 +248,7 @@ def main():
|
|
248 |
choices=language_to_models[language_choices[0]],
|
249 |
label="Select a model",
|
250 |
value=language_to_models[language_choices[0]][0],
|
|
|
251 |
)
|
252 |
decoding_method_radio = gr.Radio(
|
253 |
label="Decoding method",
|
|
|
248 |
choices=language_to_models[language_choices[0]],
|
249 |
label="Select a model",
|
250 |
value=language_to_models[language_choices[0]][0],
|
251 |
+
allow_custom_value=True
|
252 |
)
|
253 |
decoding_method_radio = gr.Radio(
|
254 |
label="Decoding method",
|
toolbox/k2_sherpa/nn_models.py
CHANGED
@@ -224,6 +224,38 @@ model_map = {
|
|
224 |
"tokens_file_sub_folder": "data/lang_bpe_500",
|
225 |
"loader": "load_sherpa_offline_recognizer",
|
226 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
227 |
],
|
228 |
"Chinese+English": [
|
229 |
{
|
|
|
224 |
"tokens_file_sub_folder": "data/lang_bpe_500",
|
225 |
"loader": "load_sherpa_offline_recognizer",
|
226 |
},
|
227 |
+
{
|
228 |
+
"repo_id": "WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02",
|
229 |
+
"nn_model_file": "cpu_jit-torch-1.10.pt",
|
230 |
+
"nn_model_file_sub_folder": "exp",
|
231 |
+
"tokens_file": "tokens.txt",
|
232 |
+
"tokens_file_sub_folder": "data/lang_bpe_500",
|
233 |
+
"loader": "load_sherpa_offline_recognizer",
|
234 |
+
},
|
235 |
+
{
|
236 |
+
"repo_id": "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14",
|
237 |
+
"nn_model_file": "cpu_jit.pt",
|
238 |
+
"nn_model_file_sub_folder": "exp",
|
239 |
+
"tokens_file": "tokens.txt",
|
240 |
+
"tokens_file_sub_folder": "data/lang_bpe_500",
|
241 |
+
"loader": "load_sherpa_offline_recognizer",
|
242 |
+
},
|
243 |
+
{
|
244 |
+
"repo_id": "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11",
|
245 |
+
"nn_model_file": "cpu_jit-torch-1.10.0.pt",
|
246 |
+
"nn_model_file_sub_folder": "exp",
|
247 |
+
"tokens_file": "tokens.txt",
|
248 |
+
"tokens_file_sub_folder": "data/lang_bpe_500",
|
249 |
+
"loader": "load_sherpa_offline_recognizer",
|
250 |
+
},
|
251 |
+
{
|
252 |
+
"repo_id": "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13",
|
253 |
+
"nn_model_file": "cpu_jit.pt",
|
254 |
+
"nn_model_file_sub_folder": "exp",
|
255 |
+
"tokens_file": "tokens.txt",
|
256 |
+
"tokens_file_sub_folder": "data/lang_bpe_500",
|
257 |
+
"loader": "load_sherpa_offline_recognizer",
|
258 |
+
},
|
259 |
],
|
260 |
"Chinese+English": [
|
261 |
{
|