wt002 commited on
Commit
e397cdc
·
verified ·
1 Parent(s): 4d61251

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -0
agent.py CHANGED
@@ -462,6 +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
 
466
  # Initialize the desired model and parameters
467
  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"