Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import yaml
|
2 |
import re
|
3 |
-
from smolagents import HfApiModel
|
4 |
from tools.final_answer import FinalAnswerTool
|
5 |
from Gradio_UI import GradioUI
|
6 |
|
@@ -53,6 +53,17 @@ if "final_answer" not in prompt_templates:
|
|
53 |
prompt_templates["final_answer"] = {"pre_messages": "", "post_messages": ""}
|
54 |
|
55 |
# Initialize CodeAgent with a low verbosity level to reduce extra debug output.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
56 |
|
57 |
def run_agent(user_input):
|
58 |
"""
|
|
|
1 |
import yaml
|
2 |
import re
|
3 |
+
from smolagents import CodeAgent, HfApiModel
|
4 |
from tools.final_answer import FinalAnswerTool
|
5 |
from Gradio_UI import GradioUI
|
6 |
|
|
|
53 |
prompt_templates["final_answer"] = {"pre_messages": "", "post_messages": ""}
|
54 |
|
55 |
# Initialize CodeAgent with a low verbosity level to reduce extra debug output.
|
56 |
+
agent = CodeAgent(
|
57 |
+
model=model,
|
58 |
+
tools=[final_answer],
|
59 |
+
max_steps=4,
|
60 |
+
verbosity_level=0,
|
61 |
+
grammar=None,
|
62 |
+
planning_interval=None,
|
63 |
+
name="Hypertension Prevention Advisor",
|
64 |
+
description=system_prompt,
|
65 |
+
prompt_templates=prompt_templates
|
66 |
+
)
|
67 |
|
68 |
def run_agent(user_input):
|
69 |
"""
|