Spaces:
Sleeping
Sleeping
update
Browse files
toolbox/k2_sherpa/nn_models.py
CHANGED
|
@@ -259,6 +259,11 @@ def load_recognizer(local_model_dir: Path,
|
|
| 259 |
decoding_method=decoding_method,
|
| 260 |
**kwargs_
|
| 261 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 262 |
else:
|
| 263 |
raise NotImplementedError("loader not support: {}".format(loader))
|
| 264 |
return recognizer
|
|
|
|
| 259 |
decoding_method=decoding_method,
|
| 260 |
**kwargs_
|
| 261 |
)
|
| 262 |
+
elif loader == "load_sherpa_offline_recognizer_from_transducer":
|
| 263 |
+
recognizer = load_sherpa_offline_recognizer_from_transducer(
|
| 264 |
+
decoding_method=decoding_method,
|
| 265 |
+
**kwargs_
|
| 266 |
+
)
|
| 267 |
else:
|
| 268 |
raise NotImplementedError("loader not support: {}".format(loader))
|
| 269 |
return recognizer
|