Psiska commited on
Commit
43da6fa
·
1 Parent(s): ae84236

Changed the use of model 4

Browse files
Files changed (1) hide show
  1. agent.py +1 -2
agent.py CHANGED
@@ -15,8 +15,7 @@ class BasicAgent:
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
  )
 
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
  )