2MoOn24mOoN4 commited on
Commit
cbd4dd8
·
verified ·
1 Parent(s): c6c2d31

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -54,7 +54,7 @@ search_tool = DuckDuckGoSearchTool()
54
  web_search = VisitWebpageTool()
55
  #=======Agent========#
56
  hf_token = os.environ.get("HF_TOKEN")
57
- model = InferenceClientModel(token = hf_token, model_id = 'xai-org/grok-1')
58
  agent = CodeAgent(
59
  tools=[
60
  multiply,
 
54
  web_search = VisitWebpageTool()
55
  #=======Agent========#
56
  hf_token = os.environ.get("HF_TOKEN")
57
+ model = InferenceClientModel(token = hf_token, model_id = 'meta-llama/Llama-2-7b-chat-hf', provider = "huggingface")
58
  agent = CodeAgent(
59
  tools=[
60
  multiply,