m-ric HF Staff commited on
Commit
54b2fe2
·
verified ·
1 Parent(s): a0a4d30

Upload tool

Browse files
Files changed (1) hide show
  1. tool.py +2 -2
tool.py CHANGED
@@ -2,9 +2,9 @@ from agents import Tool
2
 
3
  class HFModelDownloadsTool(Tool):
4
  name = "model_download_counter"
5
- description = "
6
  This is a tool that returns the most downloaded model of a given task on the Hugging Face Hub.
7
- It returns the name of the checkpoint."
8
  inputs = {"task":{"type":"string","description":"the task category (such as text-classification, depth-estimation, etc)"}}
9
  output_type = "string"
10
 
 
2
 
3
  class HFModelDownloadsTool(Tool):
4
  name = "model_download_counter"
5
+ description = """
6
  This is a tool that returns the most downloaded model of a given task on the Hugging Face Hub.
7
+ It returns the name of the checkpoint."""
8
  inputs = {"task":{"type":"string","description":"the task category (such as text-classification, depth-estimation, etc)"}}
9
  output_type = "string"
10