wt002 commited on
Commit
661a4d9
·
verified ·
1 Parent(s): e397cdc

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -462,7 +462,7 @@ tools = [wiki_tool, calc_tool, file_tool, web_tool, arvix_tool, youtube_tool, vi
462
  # Get the Hugging Face API token from the environment variable
463
  hf_token = os.getenv("HF_TOKEN")
464
 
465
- login(token="HF_TOKEN")
466
 
467
  # Initialize the desired model and parameters
468
  model_name = "mistralai/Mistral-7B-Instruct-v0.1"
 
462
  # Get the Hugging Face API token from the environment variable
463
  hf_token = os.getenv("HF_TOKEN")
464
 
465
+ login(token=hf_token)
466
 
467
  # Initialize the desired model and parameters
468
  model_name = "mistralai/Mistral-7B-Instruct-v0.1"