WaelDahech commited on
Commit
e545bf8
·
1 Parent(s): d0b24f4

Update agent configuration and tools

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,4 +1,4 @@
1
- from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
2
  import datetime
3
  import requests
4
  import pytz
@@ -85,9 +85,11 @@ TOOL_REGISTRY = [
85
  Tool(name="japanese_baseball_api", entry_point="tools.japanese_baseball_api.call"),
86
  ]
87
 
 
 
88
 
89
  agent = CodeAgent(
90
- model=model,
91
  tools=[final_answer, *TOOL_REGISTRY], ## add your tools here (don't remove final answer)
92
  max_steps=6,
93
  verbosity_level=1,
 
1
+ from smolagents import CodeAgent,DuckDuckGoSearchTool, InferenceClientModel,load_tool,tool
2
  import datetime
3
  import requests
4
  import pytz
 
85
  Tool(name="japanese_baseball_api", entry_point="tools.japanese_baseball_api.call"),
86
  ]
87
 
88
+
89
+ model_id = "Qwen/Qwen2.5-Coder-32B-Instruct"
90
 
91
  agent = CodeAgent(
92
+ model= InferenceClientModel(model_id=model_id),
93
  tools=[final_answer, *TOOL_REGISTRY], ## add your tools here (don't remove final answer)
94
  max_steps=6,
95
  verbosity_level=1,