ABVM commited on
Commit
8e91062
·
verified ·
1 Parent(s): 4bec693

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -3
agent.py CHANGED
@@ -2,13 +2,13 @@ from smolagents import CodeAgent, InferenceClientModel
2
  from smolagents import VisitWebpageTool, WebSearchTool, WikipediaSearchTool, PythonInterpreterTool,FinalAnswerTool
3
  import os
4
 
5
- os.getenv("hf_token")
6
 
7
  class GaiaAgent:
8
  def __init__(self):
9
  import os
10
- token = os.getenv("hf_token")
11
- print("DEBUG: hf_token is", token[:6]+ "...")
12
  self.model = InferenceClientModel(
13
  max_tokens = 1024,
14
  temperature = 0.5,
 
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):
9
  import os
10
+ token = os.getenv("HF_API_TOKEN")
11
+ print("DEBUG: HF_API_TOKEN is", token[:6]+ "...")
12
  self.model = InferenceClientModel(
13
  max_tokens = 1024,
14
  temperature = 0.5,