computergibs/PythonHelper

1from transformers import FSMTForConditionalGeneration, FSMTTokenizer2 3MODEL_NAME = "facebook/wmt19-en-ru"4 5# инициализируем токенизатор6tokenizer = FSMTTokenizer.from_pretrained(MODEL_NAME)7 8# инициализируем модель для генерации условий9model = FSMTForConditionalGeneration.from_pretrained(MODEL_NAME)10 11 12# определяем функцию для перевода текста13def translate_text(input_text):14    input_ids = tokenizer.encode(input_text, return_tensors="pt")15    outputs = model.generate(input_ids)16    decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)17    return decoded18