Spaces:
Sleeping
Sleeping
Changed the use of model 3
Browse files
agent.py
CHANGED
|
@@ -15,7 +15,8 @@ class BasicAgent:
|
|
| 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 |
)
|
|
|
|
| 15 |
|
| 16 |
agent = CodeAgent(
|
| 17 |
tools=[tls.search_tool, tls.calculate_cargo_travel_time],
|
| 18 |
+
#model=InferenceClientModel(),
|
| 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 |
)
|