Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
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 =
|
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(
|