Update app.py

app.py

@@ -10,7 +10,7 @@ import json
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
-class MathModelTool(BaseTool):
+class math_model_tool(BaseTool):
     name = "math_model"
     description = "Answers advanced math questions using a pretrained math model."
 
@@ -80,7 +80,7 @@ class BasicAgent:
         print("BasicAgent initialized.")
         self.agent = CodeAgent(
             model=model,
-            tools=[web_search,python_interpreter,visit_webpage_tool], 
+            tools=[web_search,python_interpreter,visit_webpage_tool,math_model_tool], 
             max_steps=2,
             verbosity_level=1,
             grammar=None,