Upload tool

app.py

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

model_downloads.py

@@ -3,12 +3,16 @@ 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 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."
     )
 
+    inputs = ['text']
+    outputs = ['text']
+
     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."
+        return f"{model.id} with {model.downloads} downloads over the past 30 days."

requirements.txt

@@ -1,2 +1,2 @@
+transformers
 huggingface_hub
-git+https://github.com/huggingface/transformers@test_composition

tool_config.json

@@ -1,5 +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": "model-download-counter",
     "tool_class": "model_downloads.HFModelDownloadsTool"
+  }
 }