update chat function

app.py

@@ -39,9 +39,15 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 # def chatbot_interface(user_input):
 #     return generate_response(user_input)
 
+def yes_man(message, history):
+    if message.endswith("?"):
+        return "Yes"
+    else:
+        return "Ask me anything!"
+
 # Create the Gradio interface
 iface = gr.Interface(
-    fn=chatbot_interface,
+    fn=yes_man,
     inputs=gr.inputs.Textbox(lines=2, placeholder="Type your message here..."),
     outputs="text",
     title="Llama-2 7B Chatbot",