Update app.py

app.py

@@ -5,22 +5,25 @@ import gradio as gr
 chatbot = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta")
 
 def chat_with_ai(message, history=[]):
-    prompt = ""
-    for user_msg, bot_msg in history:
-        prompt += f"User: {user_msg}\nAI: {bot_msg}\n"
-    prompt += f"User: {message}\nAI:"
+    messages = [
+    {
+        "role": "system",
+        "content": "You are a friendly chatbot who always responds in the style of a pirate",
+    },
+    {"role": "user", "content": "message"},
+    ]
 
     response = chatbot(
-        prompt,
+        messages,
         max_new_tokens=200,
         do_sample=True,
         temperature=0.7,
         top_p=0.9
     )
 
-    reply = response[0]["generated_text"].split("AI:")[-1].strip()
-    history.append((message, reply))
-    return reply, history
+    reply = response[0]["generated_text"]
+    
+    return reply
 
 demo = gr.ChatInterface(fn=chat_with_ai, title="Adhem AI Chatbot 🤖 (Free Model)")
 demo.launch(share=True)