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 (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),
+    "ChatGPT-like (ruGPT3small)": pipeline("text-generation", model="ai-forever/rugpt3small_based_on_gpt2", tokenizer="ai-forever/rugpt3small_based_on_gpt2", device=-1),
+    "DeepSeek-like (ruGPT3large)": pipeline("text-generation", model="ai-forever/rugpt3large_based_on_gpt2", tokenizer="ai-forever/rugpt3large_based_on_gpt2", device=-1),
+    "GigaChat-like (rubert-tiny2)": pipeline("text-generation", model="cointegrated/rubert-tiny2", tokenizer="cointegrated/rubert-tiny2", device=-1),
 }
 
 # Промпты
@@ -59,37 +59,37 @@ def generate_comparison(user_input):
         }
 
     return (
-        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"]
+        result["ChatGPT-like (ruGPT3small)"]["cot_answer"], result["ChatGPT-like (ruGPT3small)"]["cot_time"],
+        result["ChatGPT-like (ruGPT3small)"]["simple_answer"], result["ChatGPT-like (ruGPT3small)"]["simple_time"],
+        result["DeepSeek-like (ruGPT3large)"]["cot_answer"], result["DeepSeek-like (ruGPT3large)"]["cot_time"],
+        result["DeepSeek-like (ruGPT3large)"]["simple_answer"], result["DeepSeek-like (ruGPT3large)"]["simple_time"],
+        result["GigaChat-like (rubert-tiny2)"]["cot_answer"], result["GigaChat-like (rubert-tiny2)"]["cot_time"],
+        result["GigaChat-like (rubert-tiny2)"]["simple_answer"], result["GigaChat-like (rubert-tiny2)"]["simple_time"]
     )
 
 # Интерфейс
 with gr.Blocks() as demo:
-    gr.Markdown("## Сравнение моделей: ruDialoGPT, ruGPT3-small, ruGPT3-large (Классификация обращений)")
+    gr.Markdown("## Сравнение моделей: ruGPT3small, ruGPT3large, rubert-tiny2 (Классификация обращений)")
 
     inp = gr.Textbox(label="Вопрос клиента", placeholder="Например: Я не могу войти в личный кабинет", lines=2)
     btn = gr.Button("Сгенерировать")
 
     # ChatGPT-like
-    gr.Markdown("### ChatGPT-like (ruDialoGPT-medium)")
+    gr.Markdown("### ChatGPT-like (ruGPT3small)")
     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 (ruGPT3-small)")
+    gr.Markdown("### DeepSeek-like (ruGPT3large)")
     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 (ruGPT3-large)")
+    gr.Markdown("### GigaChat-like (rubert-tiny2)")
     cot3 = gr.Textbox(label="CoT ответ")
     cot3_time = gr.Textbox(label="Время CoT")
     simple3 = gr.Textbox(label="Обычный ответ")