wt002 commited on
Commit
3778af2
·
verified ·
1 Parent(s): d2e0bae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -157,9 +157,10 @@ def reasoning_node(state: AgentState) -> AgentState:
157
  # Create the LLM
158
  llm = HuggingFaceHub(
159
  repo_id="HuggingFaceH4/zephyr-7b-beta",
160
- huggingfacehub_api_token=token,
161
  model_kwargs={"temperature": 0.1, "max_new_tokens": 500}
162
  )
 
163
  chat_model = ChatHuggingFace(llm=llm)
164
 
165
  # Build prompt
 
157
  # Create the LLM
158
  llm = HuggingFaceHub(
159
  repo_id="HuggingFaceH4/zephyr-7b-beta",
160
+ huggingfacehub_api_token=os.environ["HF_TOKEN"],
161
  model_kwargs={"temperature": 0.1, "max_new_tokens": 500}
162
  )
163
+
164
  chat_model = ChatHuggingFace(llm=llm)
165
 
166
  # Build prompt