Upload tool

model_downloads.py

@@ -9,9 +9,9 @@ class HFModelDownloadsTool(Tool):
     )
 
     inputs = {
-        "name": {
+        "task": {
             "type": str,
-            "description": "the name of the category (such as text-classification, depth-estimation, etc)",
+            "description": "the task of the 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 of the category (such as text-classification, depth-estimation, etc)'}}",
   "name": "model_download_counter",
   "output_type": "<class 'str'>",
   "tool_class": "model_downloads.HFModelDownloadsTool"