Update app.py

app.py

@@ -17,7 +17,7 @@ def generate_response(model_name, message, chat_history):
         input_text = message
     response = model.generate(input_text, max_tokens=100)
     chat_history.append((input_text, response))
-    return "", chat_history, response
+    return "", chat_history
 
 # Create Gradio UI
 with gr.Blocks() as demo:
@@ -40,9 +40,8 @@ with gr.Blocks() as demo:
             chatbot = gr.Chatbot(label="Chatroom", value=[(None, "How may I help you today?")])
             
     message = gr.Textbox(label="Message")
-    state = gr.State()
 
-    message.submit(generate_response, inputs=[model_dropdown, message, state], outputs=[message, chatbot, state])
+    message.submit(generate_response, inputs=[model_dropdown, message, chatbot], outputs=[message, chatbot])
 
 # Launch the Gradio app
 demo.launch()