init

app.py

@@ -187,11 +187,19 @@ def summarizer(text, max_length, min_length, model_name):
         return f"Ошибка: {str(e)}"
 
 @measure_time_and_save("Переводчик")
-def translator(text, model_name):
+def translator(text, model_name, src_lang=None, tgt_lang=None):
     """Перевод текста"""
     try:
         translator_pipe = get_pipeline("translation", model_name)
-        result = translator_pipe(text)
+        
+        # Для mBART моделей требуются src_lang и tgt_lang
+        if "mbart" in model_name.lower():
+            if not src_lang or not tgt_lang:
+                return "Ошибка: Для модели mBART необходимо указать исходный и целевой языки"
+            result = translator_pipe(text, src_lang=src_lang, tgt_lang=tgt_lang)
+        else:
+            result = translator_pipe(text)
+        
         if isinstance(result, list):
             result = result[0]
         return result['translation_text']
@@ -685,12 +693,44 @@ with gr.Blocks(title="Трансформеры Hugging Face", theme=gr.themes.So
                             value="Helsinki-NLP/opus-mt-en-ru",
                             label="Выберите модель"
                         )
+                        translator_src_lang = gr.Dropdown(
+                            choices=[
+                                "en_XX", "ru_RU", "de_DE", "fr_XX", "es_XX", 
+                                "it_IT", "pt_XX", "ja_XX", "ko_KR", "zh_CN",
+                                "ar_AR", "hi_IN", "tr_TR", "vi_VN", "th_TH"
+                            ],
+                            value="en_XX",
+                            label="Исходный язык (для mBART)",
+                            visible=False
+                        )
+                        translator_tgt_lang = gr.Dropdown(
+                            choices=[
+                                "en_XX", "ru_RU", "de_DE", "fr_XX", "es_XX", 
+                                "it_IT", "pt_XX", "ja_XX", "ko_KR", "zh_CN",
+                                "ar_AR", "hi_IN", "tr_TR", "vi_VN", "th_TH"
+                            ],
+                            value="ru_RU",
+                            label="Целевой язык (для mBART)",
+                            visible=False
+                        )
                         translator_btn = gr.Button("Перевести", variant="primary")
                     with gr.Column():
                         translator_output = gr.Textbox(label="Перевод", lines=3)
+                
+                def update_lang_visibility(model_name):
+                    """Показывает/скрывает поля языков в зависимости от модели"""
+                    is_mbart = "mbart" in model_name.lower()
+                    return gr.update(visible=is_mbart), gr.update(visible=is_mbart)
+                
+                translator_model.change(
+                    fn=update_lang_visibility,
+                    inputs=translator_model,
+                    outputs=[translator_src_lang, translator_tgt_lang]
+                )
+                
                 translator_btn.click(
                     translator,
-                    inputs=[translator_input, translator_model],
+                    inputs=[translator_input, translator_model, translator_src_lang, translator_tgt_lang],
                     outputs=translator_output
                 )