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(
|