Update app.py

app.py

@@ -1,41 +1,17 @@
-import gradio as gr
 from transformers import pipeline
 
-# Загружаем пайплайн генерации (быстрая и лёгкая модель)
-generator = pipeline("text2text-generation", model="google/flan-t5-small")
+generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
 
-# История сообщений
 chat_history = []
 
 def chat(message):
     global chat_history
 
-    if not message.strip():
-        return chat_history + [["Ты", message], ["FlareGPT", "Пожалуйста, введите сообщение."]]
+    # Генерация текста
+    response = generator(message, max_new_tokens=100, do_sample=True)[0]['generated_text']
 
-    # Генерация ответа
-    response = generator(message, max_new_tokens=100)[0]["generated_text"]
-
-    # Добавляем в историю
+    # Добавление в историю
     chat_history.append(["Ты", message])
     chat_history.append(["FlareGPT", response])
 
     return chat_history
-
-def clear_history():
-    global chat_history
-    chat_history = []
-    return []
-
-# Интерфейс Gradio
-with gr.Blocks(title="FlareGPT Light") as demo:
-    gr.Markdown("## Поговори с FlareGPT!")
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox(label="Сообщение")
-    send_btn = gr.Button("Отправить")
-    clear_btn = gr.Button("Очистить")
-
-    send_btn.click(chat, inputs=msg, outputs=chatbot)
-    clear_btn.click(clear_history, outputs=chatbot)
-
-demo.launch()