Update app.py

app.py

@@ -1,33 +1,36 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+from transformers import pipeline
 
 # Загружаем модель
-model_name = "google/flan-t5-small"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+generator = pipeline("text2text-generation", model="google/flan-t5-small")
 
-# Пайплайн генерации текста
-generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
+# История сообщений
+chat_history = []
 
 # Функция чата
-def chat(message, history):
+def chat(message):
+    global chat_history
+
     if not message.strip():
-        return "Пожалуйста, напиши что-то!"
-    
+        return chat_history + [["Ты", message], ["FlareGPT", "Пожалуйста, напиши что-то!"]]
+
     # Генерация ответа
     response = generator(message, max_new_tokens=100)[0]["generated_text"]
-    
-    # Простой фильтр для исключения повтора
-    response = ' '.join(sorted(set(response.split()), key=response.split().index))  # Убираем дубли в ответе
 
-    return response
+    # Обновляем историю
+    chat_history.append(["Ты", message])
+    chat_history.append(["FlareGPT", response])
+
+    return chat_history
 
 # Интерфейс
-gr.Interface(
+iface = gr.Interface(
     fn=chat,
-    inputs=[gr.Textbox(placeholder="Напиши сюда...", label="Сообщение")],
-    outputs=[gr.Chatbot()],
+    inputs=gr.Textbox(label="Сообщение"),
+    outputs=gr.Chatbot(label="FlareGPT Light"),
     title="FlareGPT Light",
     description="Поговори с FlareGPT!",
-    theme="compact"
-).launch()
+    theme="default"
+)
+
+iface.launch()