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