from transformers import FSMTForConditionalGeneration, FSMTTokenizer mname = "facebook/wmt19-en-ru" tokenizer = FSMTTokenizer.from_pretrained( mname) model = FSMTForConditionalGeneration.from_pretrained( mname) 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