Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -4,7 +4,8 @@ from tools import search_tool, visit_webpage, handle_file, transcription_tool
|
|
| 4 |
|
| 5 |
# model
|
| 6 |
model = InferenceClientModel(
|
| 7 |
-
model_id="Qwen/Qwen2.5-
|
|
|
|
| 8 |
)
|
| 9 |
# load the prompt template from yaml file
|
| 10 |
with open("prompts.yaml", 'r') as stream:
|
|
|
|
| 4 |
|
| 5 |
# model
|
| 6 |
model = InferenceClientModel(
|
| 7 |
+
model_id="Qwen/Qwen2.5-72B-Instruct",
|
| 8 |
+
token=os.getenv("HF_TOKEN")
|
| 9 |
)
|
| 10 |
# load the prompt template from yaml file
|
| 11 |
with open("prompts.yaml", 'r') as stream:
|