Update app.py

app.py

@@ -2,11 +2,11 @@ import gradio as gr
 import time
 from transformers import pipeline
 
-# Настройка моделей (лёгкие и совместимые с Hugging Face Spaces)
+# Лёгкие, русскоязычные модели
 models = {
-    "ChatGPT-like (Falcon)": pipeline("text-generation", model="tiiuae/falcon-7b-instruct", tokenizer="tiiuae/falcon-7b-instruct", device=-1),
-    "DeepSeek-like": pipeline("text-generation", model="deepseek-ai/deepseek-coder-6.7b-instruct", tokenizer="deepseek-ai/deepseek-coder-6.7b-instruct", device=-1),
-    "GigaChat-like (Mistral)": pipeline("text-generation", model="mistralai/Mistral-7B-Instruct", tokenizer="mistralai/Mistral-7B-Instruct", device=-1)
+    "ChatGPT-like (ruDialoGPT-medium)": pipeline("text-generation", model="sberbank-ai/ruDialoGPT-medium", tokenizer="sberbank-ai/ruDialoGPT-medium", device=-1),
+    "DeepSeek-like (ruGPT3-small)": pipeline("text-generation", model="ai-forever/rugpt3small_based_on_gpt2", tokenizer="ai-forever/rugpt3small_based_on_gpt2", device=-1),
+    "GigaChat-like (ruGPT3-large)": pipeline("text-generation", model="ai-forever/rugpt3large_based_on_gpt2", tokenizer="ai-forever/rugpt3large_based_on_gpt2", device=-1),
 }
 
 # Промпты
@@ -59,37 +59,37 @@ def generate_comparison(user_input):
         }
 
     return (
-        result["ChatGPT-like (Falcon)"]["cot_answer"], result["ChatGPT-like (Falcon)"]["cot_time"],
-        result["ChatGPT-like (Falcon)"]["simple_answer"], result["ChatGPT-like (Falcon)"]["simple_time"],
-        result["DeepSeek-like"]["cot_answer"], result["DeepSeek-like"]["cot_time"],
-        result["DeepSeek-like"]["simple_answer"], result["DeepSeek-like"]["simple_time"],
-        result["GigaChat-like (Mistral)"]["cot_answer"], result["GigaChat-like (Mistral)"]["cot_time"],
-        result["GigaChat-like (Mistral)"]["simple_answer"], result["GigaChat-like (Mistral)"]["simple_time"]
+        result["ChatGPT-like (ruDialoGPT-medium)"]["cot_answer"], result["ChatGPT-like (ruDialoGPT-medium)"]["cot_time"],
+        result["ChatGPT-like (ruDialoGPT-medium)"]["simple_answer"], result["ChatGPT-like (ruDialoGPT-medium)"]["simple_time"],
+        result["DeepSeek-like (ruGPT3-small)"]["cot_answer"], result["DeepSeek-like (ruGPT3-small)"]["cot_time"],
+        result["DeepSeek-like (ruGPT3-small)"]["simple_answer"], result["DeepSeek-like (ruGPT3-small)"]["simple_time"],
+        result["GigaChat-like (ruGPT3-large)"]["cot_answer"], result["GigaChat-like (ruGPT3-large)"]["cot_time"],
+        result["GigaChat-like (ruGPT3-large)"]["simple_answer"], result["GigaChat-like (ruGPT3-large)"]["simple_time"]
     )
 
 # Интерфейс
 with gr.Blocks() as demo:
-    gr.Markdown("## Сравнение моделей: Falcon, DeepSeek, Mistral (Классификация обращений в банке)")
+    gr.Markdown("## Сравнение моделей: ruDialoGPT, ruGPT3-small, ruGPT3-large (Классификация обращений)")
 
     inp = gr.Textbox(label="Вопрос клиента", placeholder="Например: Я не могу войти в личный кабинет", lines=2)
     btn = gr.Button("Сгенерировать")
 
     # ChatGPT-like
-    gr.Markdown("### ChatGPT-like (Falcon)")
+    gr.Markdown("### ChatGPT-like (ruDialoGPT-medium)")
     cot1 = gr.Textbox(label="CoT ответ")
     cot1_time = gr.Textbox(label="Время CoT")
     simple1 = gr.Textbox(label="Обычный ответ")
     simple1_time = gr.Textbox(label="Время обычного")
 
     # DeepSeek-like
-    gr.Markdown("### DeepSeek-like")
+    gr.Markdown("### DeepSeek-like (ruGPT3-small)")
     cot2 = gr.Textbox(label="CoT ответ")
     cot2_time = gr.Textbox(label="Время CoT")
     simple2 = gr.Textbox(label="Обычный ответ")
     simple2_time = gr.Textbox(label="Время обычного")
 
     # GigaChat-like
-    gr.Markdown("### GigaChat-like (Mistral)")
+    gr.Markdown("### GigaChat-like (ruGPT3-large)")
     cot3 = gr.Textbox(label="CoT ответ")
     cot3_time = gr.Textbox(label="Время CoT")
     simple3 = gr.Textbox(label="Обычный ответ")