Update agents/agents_nodes.py
Browse files- agents/agents_nodes.py +1 -1
agents/agents_nodes.py
CHANGED
|
@@ -29,7 +29,7 @@ from langchain_core.messages import AIMessage
|
|
| 29 |
# Initialize Hugging Face endpoint (replace with your model)
|
| 30 |
HF_MODEL = "google/gemma-2b-it"
|
| 31 |
llm_endpoint = HuggingFaceEndpoint(
|
| 32 |
-
endpoint_url=f"https://
|
| 33 |
huggingfacehub_api_token=os.environ["HF_TOKEN"],
|
| 34 |
max_new_tokens=500,
|
| 35 |
temperature=0
|
|
|
|
| 29 |
# Initialize Hugging Face endpoint (replace with your model)
|
| 30 |
HF_MODEL = "google/gemma-2b-it"
|
| 31 |
llm_endpoint = HuggingFaceEndpoint(
|
| 32 |
+
endpoint_url=f"https://huggingface.co/models/{HF_MODEL}",
|
| 33 |
huggingfacehub_api_token=os.environ["HF_TOKEN"],
|
| 34 |
max_new_tokens=500,
|
| 35 |
temperature=0
|