Spaces:
Sleeping
Sleeping
Changed the use of model 2
Browse files
agent.py
CHANGED
|
@@ -11,11 +11,11 @@ class BasicAgent:
|
|
| 11 |
def __init__(self):
|
| 12 |
print("BasicAgent initialized.")
|
| 13 |
def __call__(self, question: str) -> str:
|
| 14 |
-
model = HfApiModel(model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud", provider="together", use_auth_token=True)
|
| 15 |
|
| 16 |
agent = CodeAgent(
|
| 17 |
tools=[tls.search_tool, tls.calculate_cargo_travel_time],
|
| 18 |
-
model=
|
| 19 |
additional_authorized_imports=["pandas"],
|
| 20 |
max_steps=20,
|
| 21 |
)
|
|
|
|
| 11 |
def __init__(self):
|
| 12 |
print("BasicAgent initialized.")
|
| 13 |
def __call__(self, question: str) -> str:
|
| 14 |
+
#model = HfApiModel(model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud", provider="together", use_auth_token=True)
|
| 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 |
)
|