ianeksdi commited on
Commit
3cb26f0
·
verified ·
1 Parent(s): 0172db2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -1
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
  """