Spaces:
Running
Running
Update agent.py
Browse files
agent.py
CHANGED
@@ -472,7 +472,7 @@ llm = HuggingFaceEndpoint(
|
|
472 |
task="text-generation",
|
473 |
huggingfacehub_api_token=os.getenv("HF_TOKEN"),
|
474 |
temperature=0.7,
|
475 |
-
|
476 |
)
|
477 |
|
478 |
# Initialize the LangChain agent with the tool(s) and the model
|
|
|
472 |
task="text-generation",
|
473 |
huggingfacehub_api_token=os.getenv("HF_TOKEN"),
|
474 |
temperature=0.7,
|
475 |
+
max_new_tokens=1024
|
476 |
)
|
477 |
|
478 |
# Initialize the LangChain agent with the tool(s) and the model
|