Chris4K commited on
Commit
19b814d
·
verified ·
1 Parent(s): 8f6fc3b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -32,7 +32,8 @@ def handle_submission():
32
  selected_tools = [tool for tool, checkbox in zip(tools, tool_checkboxes) if checkbox]
33
 
34
  # Initialize the agent with the selected tools
35
- agent = HfAgent("https://api-inference.huggingface.co/models/bigcode/starcoder", additional_tools=tools)
 
36
 
37
  # agent.config.tokenizer = tokenizer
38
  # agent.config.tools = selected_tools
 
32
  selected_tools = [tool for tool, checkbox in zip(tools, tool_checkboxes) if checkbox]
33
 
34
  # Initialize the agent with the selected tools
35
+ #agent = HfAgent("https://api-inference.huggingface.co/models/bigcode/starcoder", additional_tools=tools)
36
+ agent = HfAgent("https://api-inference.huggingface.co/models/OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5", additional_tools=tools)
37
 
38
  # agent.config.tokenizer = tokenizer
39
  # agent.config.tools = selected_tools