Update app.py

app.py

@@ -10,10 +10,16 @@ from Gradio_UI import GradioUI
 
 @tool
 def calculator(a: int, b: int) -> int:
-    "Multiply two intergers."
-    Args: a: int, b: int
-    return a * b 
+    """Multiply two intergers.
+        Args:
+        a (int): First number
+        b (int): Second number
 
+    Returns:
+        int: The product of a and b
+    """
+    return a * b
+    
 print(calculator.to_string())
        
 @tool 
@@ -57,7 +63,7 @@ from tools.final_answer import FinalAnswerTool
 from smolagents import load_tool
 
 tool_registry = {
-    "get_weather": {
+    "calculator": {
         "tool": calculator,
         "schema": calculator_tool_schema(),
         "enabled": True