Как запустить модель OpenNMT-py для перевода заданной строки в блокнот jupyter?

В документации упоминается только использование командной строки для запуска translate.py Сценарий уже дан в OpenNMT-py репо, т.е. использовать

python translate.py -model model_name.pt -src data/src-test.txt -output pred.txt

и это нормально, но как мне использовать onmt библиотека для загрузки обученной модели в блокнот Jupyter и выполнения переводов на ходу? Я хочу, чтобы это было в состоянии сделать что-то вроде этого в jupyter:

model = onmt.module.load(/path/to/model.pt)
text = "hola como estas"
model.translate(text)

Output:
>>>Hello how are you

Я не могу найти документацию о том, как это сделать. Пожалуйста помоги!

0 ответов

Другие вопросы по тегам