File size: 885 Bytes
c5f388c b1fe3e4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
# Модуль перевода языка
import transformers.pipelines.base
from transformers import pipeline
def load_text_translator_model() -> transformers.pipelines.base.Pipeline:
"""
Подгрузка модели переводчика языка
:return: Класс пайплайна для модели переводчика языка
"""
return pipeline("translation", model=f'Helsinki-NLP/opus-mt-ru-en')
def translate_to_en(text: str, translator: transformers.pipelines.base.Pipeline) -> str:
"""
Перевод текста с русского на английский
:param text: Текст
:param translator: Пайплайна для модели переводчика языка
:return: Переведенный текст
"""
text = translator(text)[0]['translation_text']
print(text)
return text
|