Psiska commited on
Commit
ae84236
·
1 Parent(s): 4a07e22

Changed the use of model 3

Browse files
Files changed (1) hide show
  1. agent.py +2 -1
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
  )