Update agents/agents_nodes.py
Browse files- agents/agents_nodes.py +2 -1
agents/agents_nodes.py
CHANGED
|
@@ -24,9 +24,10 @@ from tools.financial_tools import time_value_tool
|
|
| 24 |
import os
|
| 25 |
from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
|
| 26 |
|
|
|
|
| 27 |
|
| 28 |
llm_endpoint = HuggingFaceEndpoint(
|
| 29 |
-
|
| 30 |
huggingfacehub_api_token=os.environ["HF_TOKEN"], # Set in Space secrets
|
| 31 |
max_new_tokens=500,
|
| 32 |
temperature=0
|
|
|
|
| 24 |
import os
|
| 25 |
from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
|
| 26 |
|
| 27 |
+
repo_id="google/gemma-2b-it"
|
| 28 |
|
| 29 |
llm_endpoint = HuggingFaceEndpoint(
|
| 30 |
+
repo_id=repo_id
|
| 31 |
huggingfacehub_api_token=os.environ["HF_TOKEN"], # Set in Space secrets
|
| 32 |
max_new_tokens=500,
|
| 33 |
temperature=0
|