Update app.py

app.py

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