m-ric HF staff commited on
Commit
4e9ab94
1 Parent(s): 877353d

Update model_downloads.py

Browse files
Files changed (1) hide show
  1. model_downloads.py +1 -1
model_downloads.py CHANGED
@@ -16,6 +16,6 @@ class HFModelDownloadsTool(Tool):
16
  }
17
  output_type = "string"
18
 
19
- def __call__(self, task: str):
20
  model = next(iter(list_models(filter=task, sort="downloads", direction=-1)))
21
  return model.id
 
16
  }
17
  output_type = "string"
18
 
19
+ def forward(self, task: str):
20
  model = next(iter(list_models(filter=task, sort="downloads", direction=-1)))
21
  return model.id