Update app.py

app.py

@@ -1,7 +1,23 @@
 from transformers import pipeline
+import gradio as gr
 
+# Загружаем модель
 generator = pipeline("text2text-generation", model="google/flan-t5-small")
 
+# Функция чата
 def chat(message):
-    result = generator(message, max_new_tokens=100)[0]["generated_text"]
-    return [["Ты", message], ["FlareGPT", result]]
+    response = generator(message, max_new_tokens=100)[0]['generated_text']
+    return response
+
+# Интерфейс Gradio
+iface = gr.Interface(
+    fn=chat,
+    inputs=gr.Textbox(lines=2, placeholder="Напиши сообщение..."),
+    outputs="text",
+    title="FlareGPT Light",
+    description="Чат с лёгкой моделью на Hugging Face Space"
+)
+
+# ОБЯЗАТЕЛЬНО
+if __name__ == "__main__":
+    iface.launch()