ABVM commited on
Commit
2d69eb1
·
verified ·
1 Parent(s): 8e91062

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -1
agent.py CHANGED
@@ -2,7 +2,7 @@ from smolagents import CodeAgent, InferenceClientModel
2
  from smolagents import VisitWebpageTool, WebSearchTool, WikipediaSearchTool, PythonInterpreterTool,FinalAnswerTool
3
  import os
4
 
5
- os.getenv("HF_API_TOKEN")
6
 
7
  class GaiaAgent:
8
  def __init__(self):
@@ -15,6 +15,7 @@ class GaiaAgent:
15
  model_id = 'https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud', #"meta-llama/Llama-3.1-8B-Instruct",
16
  #custom_role_conversions=None,
17
  provider="hf-inference"
 
18
  )
19
 
20
 
 
2
  from smolagents import VisitWebpageTool, WebSearchTool, WikipediaSearchTool, PythonInterpreterTool,FinalAnswerTool
3
  import os
4
 
5
+ token=os.getenv("HF_API_TOKEN")
6
 
7
  class GaiaAgent:
8
  def __init__(self):
 
15
  model_id = 'https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud', #"meta-llama/Llama-3.1-8B-Instruct",
16
  #custom_role_conversions=None,
17
  provider="hf-inference"
18
+ token=token
19
  )
20
 
21