Spaces:
Build error
Build error
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 |
"""
|