Upload tool

model_downloads.py

@@ -2,16 +2,16 @@ from transformers import Tool
 from huggingface_hub import list_models
 
 class HFModelDownloadsTool(Tool):
-    task = "model_download_counter"
+    name = "model_download_counter"
     description = (
         "This is a tool that returns the most downloaded model of a given task on the Hugging Face Hub. "
-        "It returns the task of the checkpoint."
+        "It returns the name of the checkpoint."
     )
 
     inputs = {
-        "name": {
+        "task": {
             "type": str,
-            "description": "the name of the category (such as text-classification, depth-estimation, etc)",
+            "description": "the task category (such as text-classification, depth-estimation, etc)",
         }
     }
     output_type = str

tool_config.json

@@ -1,6 +1,6 @@
 {
   "description": "This is a tool that returns the most downloaded model of a given task on the Hugging Face Hub. It returns the name of the checkpoint.",
-  "inputs": "{'name': {'type': <class 'str'>, 'description': 'the name of the category (such as text-classification, depth-estimation, etc)'}}",
+  "inputs": "{'task': {'type': <class 'str'>, 'description': 'the task category (such as text-classification, depth-estimation, etc)'}}",
   "name": "model_download_counter",
   "output_type": "<class 'str'>",
   "tool_class": "model_downloads.HFModelDownloadsTool"