m-ric HF staff commited on
Commit
d274b8b
·
verified ·
1 Parent(s): 4a5ea51

Update model_downloads.py

Browse files
Files changed (1) hide show
  1. model_downloads.py +2 -2
model_downloads.py CHANGED
@@ -10,11 +10,11 @@ class HFModelDownloadsTool(Tool):
10
 
11
  inputs = {
12
  "task": {
13
- "type": str,
14
  "description": "the task category (such as text-classification, depth-estimation, etc)",
15
  }
16
  }
17
- output_type = str
18
 
19
  def __call__(self, task: str):
20
  model = next(iter(list_models(filter=task, sort="downloads", direction=-1)))
 
10
 
11
  inputs = {
12
  "task": {
13
+ "type": "sring",
14
  "description": "the task category (such as text-classification, depth-estimation, etc)",
15
  }
16
  }
17
+ output_type = "string"
18
 
19
  def __call__(self, task: str):
20
  model = next(iter(list_models(filter=task, sort="downloads", direction=-1)))