Update app.py

app.py

@@ -74,11 +74,6 @@ def update_chatbot(message, chat_history):
 
     chat_history = call_nvidia_api(chat_history, max_tokens_val, temperature_val, top_p_val)
     return chat_history
-    
-def chatbot_function(message, history, system_message, max_tokens, temperature, top_p):
-    return update_chatbot(message, history, system_message, max_tokens, temperature, top_p)
-
-
 
 # Gradio interface components
 system_msg = gr.Textbox(BASE_SYSTEM_MESSAGE, label="System Message", placeholder="System prompt.", lines=5)
@@ -89,8 +84,8 @@ top_p = gr.Slider(0.0, 1.0, label="Top P", step=0.05, value=0.7)
 with gr.Blocks() as demo:
     chat_history_state = gr.State([])
     chatbot = gr.ChatInterface(
-        fn=chatbot_function,   
-        additional_inputs=[system_msg, max_tokens, temperature, top_p],
+        fn=update_chatbot,
+        additional_inputs=[chat_history_state, system_msg, max_tokens, temperature, top_p],
         title="LLAMA 70B Free Demo",
         description="""
             <div style="text-align: center; font-size: 1.5em; margin-bottom: 20px;">