borisyich commited on
Commit
969a6a0
·
verified ·
1 Parent(s): 7e0b4a4

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -2
agent.py CHANGED
@@ -2,12 +2,13 @@ import os
2
  from smolagents import (InferenceClientModel, CodeAgent, ToolCallingAgent,
3
  DuckDuckGoSearchTool, VisitWebpageTool, FinalAnswerTool,
4
  WikipediaSearchTool, PythonInterpreterTool,
 
5
  )
6
 
7
  model_id = "deepseek-ai/DeepSeek-R1-0528"
8
- model = InferenceClientModel(
9
  model_id,
10
- token=os.getenv('HF_TOKEN')
11
  )
12
 
13
  web_agent = ToolCallingAgent(
 
2
  from smolagents import (InferenceClientModel, CodeAgent, ToolCallingAgent,
3
  DuckDuckGoSearchTool, VisitWebpageTool, FinalAnswerTool,
4
  WikipediaSearchTool, PythonInterpreterTool,
5
+ TransformersModel
6
  )
7
 
8
  model_id = "deepseek-ai/DeepSeek-R1-0528"
9
+ model = TransformersModel(
10
  model_id,
11
+ #token=os.getenv('HF_TOKEN')
12
  )
13
 
14
  web_agent = ToolCallingAgent(