cpatino10 commited on
Commit
ea9859d
·
verified ·
1 Parent(s): 1cd76ec

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +4 -2
agent.py CHANGED
@@ -1,4 +1,4 @@
1
- from smolagents import CodeAgent, InferenceClientModel
2
  from tools import all_tools
3
 
4
  # define GAIA specific system prompt
@@ -12,7 +12,9 @@ If you are asked for a comma separated list, apply the above rules depending of
12
  Return ONLY the final answer as the output of your last step."""
13
 
14
  # model
15
- model = InferenceClientModel()
 
 
16
 
17
  # Initialize agent
18
  agent = CodeAgent(
 
1
+ from smolagents import CodeAgent, InferenceClientModel, HfApiModel
2
  from tools import all_tools
3
 
4
  # define GAIA specific system prompt
 
12
  Return ONLY the final answer as the output of your last step."""
13
 
14
  # model
15
+ model = HfApiModel(
16
+ model_id="Qwen/Qwen2.5-Coder-1.5B-Instruct"
17
+ )
18
 
19
  # Initialize agent
20
  agent = CodeAgent(