m-ric HF Staff commited on
Commit
987ce08
·
verified ·
1 Parent(s): 40978fd

Upload tool

Browse files
Files changed (1) hide show
  1. model_downloads.py +2 -2
model_downloads.py CHANGED
@@ -2,10 +2,10 @@ from transformers import Tool
2
  from huggingface_hub import list_models
3
 
4
  class HFModelDownloadsTool(Tool):
5
- name = "model_download_counter"
6
  description = (
7
  "This is a tool that returns the most downloaded model of a given task on the Hugging Face Hub. "
8
- "It returns the name of the checkpoint."
9
  )
10
 
11
  inputs = {
 
2
  from huggingface_hub import list_models
3
 
4
  class HFModelDownloadsTool(Tool):
5
+ task = "model_download_counter"
6
  description = (
7
  "This is a tool that returns the most downloaded model of a given task on the Hugging Face Hub. "
8
+ "It returns the task of the checkpoint."
9
  )
10
 
11
  inputs = {