Upload tool

app.py

@@ -0,0 +1,4 @@
+from transformers import launch_gradio_demo
+from model_downloads import HFModelDownloadsTool
+
+launch_gradio_demo(HFModelDownloadsTool)

model_downloads.py

@@ -0,0 +1,20 @@
+from transformers import Tool
+from huggingface_hub import list_models
+
+class HFModelDownloadsTool(Tool):
+    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 name of the checkpoint."""
+
+    inputs = {
+        "task": {
+            "type": "string",
+            "description": "the task category (such as text-classification, depth-estimation, etc)",
+        }
+    }
+    output_type = "string"
+
+    def forward(self, task: str):
+        model = next(iter(list_models(filter=task, sort="downloads", direction=-1)))
+        return model.id

requirements.txt

@@ -0,0 +1,2 @@
+huggingface_hub
+transformers

tool_config.json

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