Chris4K commited on
Commit
4c6c6ab
·
verified ·
1 Parent(s): dfe0aa8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -93,10 +93,10 @@ def handle_submission():
93
  #selected_tools = [tool for tool, checkbox in tool_checkboxes]
94
 
95
  # Initialize the agent
96
- agent = CustomHfAgent(url_endpoint="https://api-inference.huggingface.co/models/bigcode/starcoder", token=os.environ['HF_token'])
97
 
98
  # Run the agent with the user's message and selected tools
99
- response = agent.run(message, tools=selected_tools)
100
  #response = agent.chat(message)
101
 
102
  print(response)
 
93
  #selected_tools = [tool for tool, checkbox in tool_checkboxes]
94
 
95
  # Initialize the agent
96
+ agent = CustomHfAgent(url_endpoint="https://api-inference.huggingface.co/models/bigcode/starcoder", token=os.environ['HF_token'], additional_tools=selected_tools)
97
 
98
  # Run the agent with the user's message and selected tools
99
+ response = agent.run(message)
100
  #response = agent.chat(message)
101
 
102
  print(response)