Chris4K commited on
Commit
a9953ef
·
verified ·
1 Parent(s): 4da4c03

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -77,10 +77,10 @@ def handle_submission():
77
  # Get the user's message and the selected tools
78
  message = st.text_input("Enter your message:", "")
79
 
80
- selected_tools = []
81
- for tool, checkbox in tool_checkboxes:
82
- if checkbox:
83
- selected_tools.append(tool)
84
 
85
  #selected_tools = [tool for tool, checkbox in tool_checkboxes]
86
 
@@ -88,7 +88,8 @@ def handle_submission():
88
  agent = CustomHfAgent(url_endpoint="https://api-inference.huggingface.co/models/bigcode/starcoder", token=os.environ['HF_token'])
89
 
90
  # Run the agent with the user's message and selected tools
91
- response = agent.run(message, tools=selected_tools)
 
92
 
93
  # Display the agent's response
94
  # Display the agent's response
 
77
  # Get the user's message and the selected tools
78
  message = st.text_input("Enter your message:", "")
79
 
80
+ # selected_tools = []
81
+ # for tool, checkbox in tool_checkboxes:
82
+ # if checkbox:
83
+ # selected_tools.append(tool)
84
 
85
  #selected_tools = [tool for tool, checkbox in tool_checkboxes]
86
 
 
88
  agent = CustomHfAgent(url_endpoint="https://api-inference.huggingface.co/models/bigcode/starcoder", token=os.environ['HF_token'])
89
 
90
  # Run the agent with the user's message and selected tools
91
+ #response = agent.run(message, tools=selected_tools)
92
+ response = agent.run(message)
93
 
94
  # Display the agent's response
95
  # Display the agent's response