Spaces:
Running
Running
Update agent.py
Browse files
agent.py
CHANGED
@@ -468,10 +468,12 @@ from langchain_huggingface import HuggingFaceEndpoint
|
|
468 |
|
469 |
# Initialize the HuggingFaceEndpoint with the desired model and parameters
|
470 |
llm = HuggingFaceEndpoint(
|
471 |
-
repo_id="
|
472 |
task="text-generation",
|
473 |
-
huggingfacehub_api_token=
|
474 |
-
|
|
|
|
|
475 |
)
|
476 |
|
477 |
|
|
|
468 |
|
469 |
# Initialize the HuggingFaceEndpoint with the desired model and parameters
|
470 |
llm = HuggingFaceEndpoint(
|
471 |
+
repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
|
472 |
task="text-generation",
|
473 |
+
huggingfacehub_api_token=os.getenv("HF_TOKEN"),
|
474 |
+
temperature=0.7,
|
475 |
+
model_kwargs={"max_length": 1024}
|
476 |
+
)
|
477 |
)
|
478 |
|
479 |
|