Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -111,8 +111,8 @@ def build_graph(provider: str = "huggingface"):
|
|
| 111 |
# TODO: Add huggingface endpoint
|
| 112 |
llm = ChatHuggingFace(
|
| 113 |
llm=HuggingFaceEndpoint(
|
| 114 |
-
|
| 115 |
-
temperature
|
| 116 |
huggingfacehub_api_token=hf_api_key
|
| 117 |
),
|
| 118 |
)
|
|
|
|
| 111 |
# TODO: Add huggingface endpoint
|
| 112 |
llm = ChatHuggingFace(
|
| 113 |
llm=HuggingFaceEndpoint(
|
| 114 |
+
endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
|
| 115 |
+
model_kwargs={"temperature": 0},
|
| 116 |
huggingfacehub_api_token=hf_api_key
|
| 117 |
),
|
| 118 |
)
|