Update app.py

app.py

@@ -9,14 +9,14 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def get_local_name_of_animal(animal_category:str)-> str:
-    """A tool that takes animal category as input and return local name of that input category of animal.
+def get_local_name_of_animal(category: str)-> str:
+    """A tool that fetches the local name of animal given the category.
     Args:
-        animal_category: category of animal
+        category: a string representing category of animal.
     """
-    if animal_category == 'dog':
+    if category == 'dog':
         return 'Jatin'
-    elif animal_category == 'cat':
+    elif category == 'cat':
         return 'Bobo'
     else:
         return 'Bareja'