Spaces:
Sleeping
Sleeping
update
Browse files
toolbox/k2_sherpa/nn_models.py
CHANGED
|
@@ -379,6 +379,10 @@ def load_recognizer(local_model_dir: Path,
|
|
| 379 |
decoding_method=decoding_method,
|
| 380 |
**kwargs_
|
| 381 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 382 |
else:
|
| 383 |
raise NotImplementedError("loader not support: {}".format(loader))
|
| 384 |
return recognizer
|
|
|
|
| 379 |
decoding_method=decoding_method,
|
| 380 |
**kwargs_
|
| 381 |
)
|
| 382 |
+
elif loader == "load_sherpa_offline_recognizer_from_whisper":
|
| 383 |
+
recognizer = load_sherpa_offline_recognizer_from_whisper(
|
| 384 |
+
**kwargs_
|
| 385 |
+
)
|
| 386 |
else:
|
| 387 |
raise NotImplementedError("loader not support: {}".format(loader))
|
| 388 |
return recognizer
|