Upload tool

app.py

@@ -0,0 +1,3 @@
+from transformers.tools.base import launch_gradio_demo from __main__ import HFModelDownloadsTool
+
+launch_gradio_demo(HFModelDownloadsTool)

model_downloads.py

@@ -0,0 +1,14 @@
+from transformers.tools import Tool
+from huggingface_hub import list_models
+
+
+class HFModelDownloadsTool(Tool):
+    description = (
+        "This is a tool that returns the most downloaded model of a given task on the Hugging Face Hub. " 
+        "It takes the name of the category (such as text-classification, depth-estimation, etc), and " 
+        "returns the name of the checkpoint alongside its 30-day download numbers."
+    )
+
+    def __call__(self, task: str):
+        model = next(iter(list_models(filter=task, sort='downloads', direction=-1)))
+        return f"{model.id} with {model.downloads} downloads over the past 30 days."

requirements.txt

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

tool_config.json

@@ -0,0 +1,7 @@
+{
+  "h_f_model_downloads": {
+    "description": "This is a tool that returns the most downloaded model of a given task on the Hugging Face Hub. It takes the name of the category (such as text-classification, depth-estimation, etc), and returns the name of the checkpoint alongside its 30-day download numbers.",
+    "name": "",
+    "tool_class": "__main__.HFModelDownloadsTool"
+  }
+}