magdap116 commited on
Commit
40f13f3
·
verified ·
1 Parent(s): 496166b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ import hashlib
8
  import json
9
  from transformers import AutoTokenizer, AutoModelForCausalLM
10
  import torch
11
-
12
 
13
  # (Keep Constants as is)
14
  # --- Constants ---
@@ -58,7 +58,7 @@ class BasicAgent:
58
  self.agent = CodeAgent(
59
  model=model,
60
  tools=[web_search,python_interpreter,visit_webpage_tool,math_model_tool],
61
- max_steps=2,
62
  verbosity_level=1,
63
  grammar=None,
64
  planning_interval=3,
 
8
  import json
9
  from transformers import AutoTokenizer, AutoModelForCausalLM
10
  import torch
11
+ from tooling import ModelMathTool
12
 
13
  # (Keep Constants as is)
14
  # --- Constants ---
 
58
  self.agent = CodeAgent(
59
  model=model,
60
  tools=[web_search,python_interpreter,visit_webpage_tool,math_model_tool],
61
+ max_steps=1,
62
  verbosity_level=1,
63
  grammar=None,
64
  planning_interval=3,