wlchee commited on
Commit
93c2c8b
·
verified ·
1 Parent(s): 920cf9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -5
app.py CHANGED
@@ -71,14 +71,18 @@ class WikipediaTool(Tool):
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
 
84
  self.tools = [
 
71
 
72
  class BasicAgent:
73
  def __init__(self):
 
74
  self.model = HfApiModel(
75
+ model_id='HuggingFaceH4/zephyr-7b-beta',
76
  max_tokens=2096,
77
+ temperature=0.3
 
 
 
78
  )
79
+ #HF_TOKEN = os.getenv("AGENT_TOKEN")
80
+ #self.model = HfApiModel(
81
+ # max_tokens=2096,
82
+ # temperature=0.3,
83
+ ##custom_role_conversions=None,
84
+ #token=HF_TOKEN
85
+ #)
86
 
87
 
88
  self.tools = [