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