Update app.py

app.py

@@ -1,21 +1,20 @@
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 import gradio as gr
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 
-# Лёгкая модель, которая хорошо работает с seq2seq
+# Используем Flan-T5 для адекватных ответов
 model_name = "google/flan-t5-small"
-
-# Загрузка токенизатора и модели
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
-# Создание пайплайна
+# Создаём пайплайн
 pipe = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
 
-# Функция ответа
-def answer(message, history):
-    result = pipe(message, max_new_tokens=100)[0]["generated_text"]
-    return result
-
-# Интерфейс Gradio
-gr.ChatInterface(fn=answer, title="FlareGPT Light").launch()
+# Логика обработки сообщений
+def respond(message, history):
+    if not message.strip():
+        return "Напиши что-нибудь!"
+    response = pipe(message, max_new_tokens=100)[0]["generated_text"]
+    return response
 
+# Интерфейс чата
+gr.ChatInterface(fn=respond, title="FlareGPT Light").launch()