Update app.py

app.py

@@ -1,8 +1,15 @@
 import gradio as gr
+from transformers import pipeline
 
-def chat(message, history):
-    return "Hello! This is BIT-GPT 0.2.8 responding."
+# Load the model
+chatbot = pipeline("text-generation", model="facebook/blenderbot-400M-distill")
 
-iface = gr.ChatInterface(chat)
+def chat_response(message):
+    response = chatbot(message, max_length=100, do_sample=True)
+    return response[0]['generated_text']
 
+# Create Gradio interface
+iface = gr.Interface(fn=chat_response, inputs="text", outputs="text", title="Bit GPT 0.2.8")
+
+# Launch app
 iface.launch()