Spaces:

patrickvonplaten
/

hf-model-downloads

Runtime error

patrickvonplaten commited on May 9, 2023

Commit

8477702

1 Parent(s): a59fb21

Update model_downloads.py

Files changed (1) hide show

model_downloads.py CHANGED Viewed

@@ -15,3 +15,4 @@ class HFModelDownloadsTool(Tool):
     def __call__(self, task: str):
         model = next(iter(list_models(filter=task, sort="downloads", direction=-1)))

     def __call__(self, task: str):
         model = next(iter(list_models(filter=task, sort="downloads", direction=-1)))
+        return model.id