utils/translators/en_ru/en_ru.py

from transformers import FSMTForConditionalGeneration, FSMTTokenizer
mname = "facebook/wmt19-en-ru"
tokenizer = FSMTTokenizer.from_pretrained(mname, cache_dir='./cache')
model = FSMTForConditionalGeneration.from_pretrained(
    mname, cache_dir='./cache')


def translator_en_ru(text):

    input_ids = tokenizer.encode(text, return_tensors="pt")
    outputs = model.generate(input_ids)
    decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)

    return decoded