math tool

setting up math tool

app.py

@@ -3,15 +3,24 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import DuckDuckGoSearchTool,HfApiModel,load_tool, CodeAgent,PythonInterpreterTool,VisitWebpageTool,BaseTool
+from smolagents import DuckDuckGoSearchTool,HfApiModel,load_tool, CodeAgent,PythonInterpreterTool,VisitWebpageTool,Tool
 import hashlib
 import json
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
-class math_model_tool(BaseTool):
+class ModelMathTool(Tool):
     name = "math_model"
     description = "Answers advanced math questions using a pretrained math model."
+    
+    inputs = {
+        "problem": {
+            "type": "string",
+            "description": "Math problem to solve.",
+        }
+    }
+
+    output_type = "string"
 
     def __init__(self, model_id="Qwen/Qwen2.5-Math-7B"):
         print(f"Loading math model: {model_id}")
@@ -22,8 +31,11 @@ class math_model_tool(BaseTool):
             device_map="auto",
             trust_remote_code=True
         )
+
+
+            
     
-    def run(self, question: str) -> str:
+    def forward(self, question: str) -> str:
         print(f"[MathModelTool] Question: {question}")
         inputs = self.tokenizer(question, return_tensors="pt").to(self.model.device)
         with torch.no_grad():
@@ -45,7 +57,7 @@ print("Made dir")
 web_search = DuckDuckGoSearchTool()
 python_interpreter = PythonInterpreterTool()
 visit_webpage_tool = VisitWebpageTool()
-
+model_math_tool = ModelMathTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'