Update app.py

app.py

@@ -1,20 +1,25 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 
-# Используем Flan-T5 для адекватных ответов
+# Загружаем flan-t5-small, а не causalLM!
 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)
+# Пайплайн генерации
+generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
 
-# Логика обработки сообщений
-def respond(message, history):
+# Функция ответа
+def chat(message, history):
     if not message.strip():
-        return "Напиши что-нибудь!"
-    response = pipe(message, max_new_tokens=100)[0]["generated_text"]
+        return "Напиши что-нибудь 😅"
+    response = generator(message, max_new_tokens=100)[0]["generated_text"]
     return response
 
-# Интерфейс чата
-gr.ChatInterface(fn=respond, title="FlareGPT Light").launch()
+# Интерфейс FlareGPT Light
+gr.ChatInterface(
+    fn=chat,
+    title="🔥 FlareGPT Light",
+    description="Лёгкий ИИ-чат на FLAN-T5-small — быстро, просто и умно.",
+    theme="compact"
+).launch()