Update app.py

app.py

@@ -12,7 +12,6 @@ import torch
 class ModelMathTool(Tool):
     name = "math_model"
     description = "Answers advanced math questions using a pretrained math model."
-    
     inputs = {
         "problem": {
             "type": "string",
@@ -35,7 +34,7 @@ class ModelMathTool(Tool):
 
             
     
-    def forward(self, question: str) -> str:
+    def forward(self, problem: str) -> str:
         print(f"[MathModelTool] Question: {question}")
         inputs = self.tokenizer(question, return_tensors="pt").to(self.model.device)
         with torch.no_grad():