Update agent.py
Browse files
agent.py
CHANGED
@@ -182,6 +182,7 @@ def build_graph(provider: str = "huggingface"):
|
|
182 |
llm=HuggingFaceEndpoint(
|
183 |
endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
|
184 |
temperature=0.2,
|
|
|
185 |
),
|
186 |
)
|
187 |
else:
|
|
|
182 |
llm=HuggingFaceEndpoint(
|
183 |
endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
|
184 |
temperature=0.2,
|
185 |
+
token="HF_TOKEN"
|
186 |
),
|
187 |
)
|
188 |
else:
|