Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
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
|