Update app.py

app.py

@@ -3,24 +3,16 @@ import gradio as gr
 from transformers import pipeline
 
 # 2. Загрузка двух нейросетей
-# ИСПОЛЬЗУЕМ БОЛЕЕ МАЛЕНЬКИЕ МОДЕЛИ
 generator = pipeline("text2text-generation", model="google/flan-t5-small")
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ru-en")
 
 # 3. Создание функции для генерации ответа
 def generate_response(prompt, language, temperature, max_length, history):
-    """
-    Эта функция принимает твой текст, переводит его, получает ответ и возвращает обновлённую историю.
-    """
-    # Так как модель "opus-mt-ru-en" работает только с русским и английским,
-    # мы будем использовать только эти языки.
     source_lang_code = "ru" if language == "ru" else "en"
     target_lang_code = "ru" if language == "ru" else "en"
     
-    # Перевод на английский
     english_prompt = translator(prompt, src_lang=source_lang_code, tgt_lang="en")[0]['translation_text']
     
-    # Получаем ответ от Flan-T5 на английском
     response_en = generator(
         english_prompt, 
         max_length=max_length, 
@@ -30,10 +22,8 @@ def generate_response(prompt, language, temperature, max_length, history):
         no_repeat_ngram_size=2  
     )[0]['generated_text']
 
-    # Перевод ответа обратно на твой язык
     final_response = translator(response_en, src_lang="en", tgt_lang=target_lang_code)[0]['translation_text']
 
-    # Добавляем в историю твой вопрос и ответ бота
     history.append([prompt, final_response])
     
     return history
@@ -45,7 +35,7 @@ with gr.Blocks(theme="soft", title="COLIN") as iface:
     
     with gr.Row():
         language_dropdown = gr.Dropdown(
-            ["en", "ru"],  # Убрали другие языки
+            ["en", "ru"],
             value="en",
             label="Select Language"
         )
@@ -62,7 +52,6 @@ with gr.Blocks(theme="soft", title="COLIN") as iface:
         clear_btn = gr.Button("New Chat")
         submit_btn = gr.Button("Submit")
 
-    # Привязка кнопок к функциям
     submit_btn.click(
         fn=generate_response,
         inputs=[textbox, language_dropdown, temperature_slider, max_length_slider, chatbot],