Spaces:
Build error
Build error
Changes HfApiModel to InferenceClientModel
Browse files
agent.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
-
from smolagents import CodeAgent, DuckDuckGoSearchTool,
|
2 |
|
3 |
# Initialize a model (use your preferred model_id)
|
4 |
-
model =
|
5 |
|
6 |
# Define custom tools if needed (optional)
|
7 |
from smolagents.tool import tool
|
|
|
1 |
+
from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel
|
2 |
|
3 |
# Initialize a model (use your preferred model_id)
|
4 |
+
model = InferenceClientModel(model_id="meta-llama/Llama-2-7b-chat-hf") # Or another model
|
5 |
|
6 |
# Define custom tools if needed (optional)
|
7 |
from smolagents.tool import tool
|