Spaces:
Sleeping
Sleeping
Changed the use of model 4
Browse files
agent.py
CHANGED
|
@@ -15,8 +15,7 @@ class BasicAgent:
|
|
| 15 |
|
| 16 |
agent = CodeAgent(
|
| 17 |
tools=[tls.search_tool, tls.calculate_cargo_travel_time],
|
| 18 |
-
|
| 19 |
-
model = HfApiModel(model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud", provider="together", use_auth_token=True),
|
| 20 |
additional_authorized_imports=["pandas"],
|
| 21 |
max_steps=20,
|
| 22 |
)
|
|
|
|
| 15 |
|
| 16 |
agent = CodeAgent(
|
| 17 |
tools=[tls.search_tool, tls.calculate_cargo_travel_time],
|
| 18 |
+
model=InferenceClientModel(),
|
|
|
|
| 19 |
additional_authorized_imports=["pandas"],
|
| 20 |
max_steps=20,
|
| 21 |
)
|