Update README.md

README.md

@@ -35,6 +35,89 @@ tags:
 | Tatoeba-test.pol-rus.pol.rus 	| 49.6 	| 0.680 |
 
 
+## Run the model
+
+```python
+from transformers import MarianMTModel, MarianTokenizer
+
+# Пути к модели и токенизатору
+model_path = "trained_model"
+
+# Загрузка модели и токенизатора
+tokenizer = MarianTokenizer.from_pretrained(model_path)
+model = MarianMTModel.from_pretrained(model_path)
+
+# Словарь, который соответствует целевым языковым маркерам для разных языков
+language_map = {
+    "bel": "bel",
+    "bel_Latn": "bel_Latn",
+    "bos_Latn": "bos_Latn",
+    "bul": "bul",
+    "bul_Latn": "bul_Latn",
+    "ces": "ces",
+    "dsb": "dsb",
+    "hrv": "hrv",
+    "hsb": "hsb",
+    "mkd": "mkd",
+    "orv_Cyrl": "orv_Cyrl",
+    "pol": "pol",
+    "rus": "rus",
+    "slv": "slv",
+    "srp_Cyrl": "srp_Cyrl",
+    "srp_Latn": "srp_Latn",
+    "ukr": "ukr"
+}
+
+# Функция для перевода с несколькими вариантами
+def translate_text(source_text, target_language, num_translations=3):
+    if target_language not in language_map:
+        print("Неверное направление перевода. Попробуйте снова.")
+        return None
+    
+    # Формируем текст с маркером целевого языка
+    language_token = f">>{target_language}<< "
+    text_with_token = language_token + source_text
+
+    # Токенизация текста
+    inputs = tokenizer(text_with_token, return_tensors="pt")
+
+    # Перевод текста с несколькими вариантами
+    translated_tokens = model.generate(
+        **inputs,
+        num_return_sequences=num_translations,  # Количество вариантов перевода
+        num_beams=num_translations  # Используем несколько beam для разнообразия
+    )
+
+    # Декодирование переведенного текста
+    translations = [tokenizer.decode(tokens, skip_special_tokens=True) for tokens in translated_tokens]
+    return translations
+
+# Основной цикл для ввода текста и вывода вариантов перевода
+print("Введите фразу для перевода или !q для выхода.")
+
+while True:
+    # Ввод фразы
+    source_text = input("Введите фразу: ")
+    
+    # Проверка на команду выхода
+    if source_text == "!q":
+        print("Выход из программы.")
+        break
+    
+    # Указание целевого языка (по умолчанию "rus")
+    target_language = "rus"
+
+    # Перевод фразы с несколькими вариантами
+    translations = translate_text(source_text, target_language)
+    
+    if translations:
+        # Вывод всех вариантов перевода
+        for idx, translation in enumerate(translations, 1):
+            print(f"Вариант {idx}: {translation}")
+
+```
+
+
 
 ### System Info: 
 - hf_name: sla-sla