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

Changed the use of model 2

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