Update app.py

app.py

@@ -19,12 +19,12 @@ model = AutoModelForCausalLM.from_pretrained(
 
 # Chat function
 @spaces.GPU
-def chat_with_bot(user_input, history):
+def chat_with_bot(message, history):
     history = history or []
     prompt = ""
     for user, bot in history:
         prompt += f"User: {user}\nAssistant: {bot}\n"
-    prompt += f"User: {user_input}\nAssistant:"
+    prompt += f"User: {message}\nAssistant:"
 
     inputs = tokenizer(prompt, return_tensors="pt").to(device)
     outputs = model.generate(
@@ -38,14 +38,13 @@ def chat_with_bot(user_input, history):
 
     decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)
     response = decoded[len(prompt):].strip().split("\n")[0]
-
-    history.append((user_input, response))
+    history.append((message, response))
     return response, history
 
-# Gradio UI
+# ✅ gr.ChatInterface expects fn(message, history) → (response, updated_history)
 gr.ChatInterface(
     fn=chat_with_bot,
-    title="Phi-2 Chatbot (ZeroGPU Safe)",
+    title="Phi-2 Chatbot (CPU-friendly)",
     theme="soft",
-    examples=["What is AI?", "Summarize the French Revolution.", "Tell me a space fact."]
+    examples=["What is AI?", "Tell me a fun fact about space.", "Summarize photosynthesis."],
 ).launch(share=True)