m-ric HF Staff commited on
Commit
018b90b
·
verified ·
1 Parent(s): a96ffa0

Upload tool

Browse files
Files changed (1) hide show
  1. model_downloads.py +2 -2
model_downloads.py CHANGED
@@ -9,9 +9,9 @@ class HFModelDownloadsTool(Tool):
9
  )
10
 
11
  inputs = {
12
- "task": {
13
  "type": str,
14
- "description": "the task of the category (such as text-classification, depth-estimation, etc)",
15
  }
16
  }
17
  output_type = str
 
9
  )
10
 
11
  inputs = {
12
+ "name": {
13
  "type": str,
14
+ "description": "the name of the category (such as text-classification, depth-estimation, etc)",
15
  }
16
  }
17
  output_type = str