Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from transformers import BertModel, BertTokenizer
+#from transformers import BertModel, BertTokenizer
 from transformers import HfAgent, load_tool
 
 # Load tools
@@ -41,7 +41,7 @@ def handle_submission():
    # outputs = agent(inputs['input_ids'], attention_mask=inputs['attention_mask'])
 
     # Display the agent's response
-    response = agent.run(inputs['input_ids'])
+    response = agent.run(message)
     st.text(f"{response:.4f}")
 
 # Add the callback function to the Streamlit app