wlchee commited on
Commit
920cf9f
·
verified ·
1 Parent(s): 7e121e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -71,13 +71,13 @@ class WikipediaTool(Tool):
71
 
72
  class BasicAgent:
73
  def __init__(self):
74
- HF_TOKEN = os.getenv("HF_TOKEN", "AGENT_TOKEN")
75
  self.model = HfApiModel(
76
  max_tokens=2096,
77
  temperature=0.3,
78
  model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
79
  custom_role_conversions=None,
80
- hf_token=HF_TOKEN
81
  )
82
 
83
 
 
71
 
72
  class BasicAgent:
73
  def __init__(self):
74
+ HF_TOKEN = os.getenv("AGENT_TOKEN")
75
  self.model = HfApiModel(
76
  max_tokens=2096,
77
  temperature=0.3,
78
  model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
79
  custom_role_conversions=None,
80
+ token=HF_TOKEN
81
  )
82
 
83