Update app.py

app.py

@@ -77,10 +77,17 @@ max_tokens = gr.Slider(20, 1024, label="Max Tokens", step=20, value=1024)
 temperature = gr.Slider(0.0, 1.0, label="Temperature", step=0.1, value=0.2)
 top_p = gr.Slider(0.0, 1.0, label="Top P", step=0.05, value=0.7)
 
+def chatbot_submit(message, chat_history, system_message, max_tokens_val, temperature_val, top_p_val):
+    """Submits the user message to the chatbot and updates the chat history."""
+    if chat_history is None:
+        chat_history = []
+    chat_history = update_chatbot(message, chat_history)
+    return chat_history
+
 with gr.Blocks() as demo:
     chat_history_state = gr.State([])
     chatbot = gr.ChatInterface(
-        fn=update_chatbot,
+        fn=chatbot_submit,
         additional_inputs=[chat_history_state, system_msg, max_tokens, temperature, top_p],
         title="LLAMA 70B Free Demo",
         description="""
@@ -107,5 +114,4 @@ with gr.Blocks() as demo:
         chatbot.textbox.value = ""
 
     chatbot.clear()
-
 demo.launch()