AbdullahRasul commited on
Commit
8a3102a
·
verified ·
1 Parent(s): 004d753

Added llama

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -22,7 +22,10 @@ class BasicAgent:
22
  raise ValueError("HF_API_TOKEN environment variable not set.")
23
 
24
  # Use the token in the HfApiModel
25
- model = HfApiModel(token=hf_token)
 
 
 
26
 
27
  # Initialize agent with tools
28
  search_tool = DuckDuckGoSearchTool()
 
22
  raise ValueError("HF_API_TOKEN environment variable not set.")
23
 
24
  # Use the token in the HfApiModel
25
+ model = HfApiModel(
26
+ model_id="meta-llama/Llama-3.1-8B-Instruct",
27
+ token=hf_token
28
+ )
29
 
30
  # Initialize agent with tools
31
  search_tool = DuckDuckGoSearchTool()