cfcooney commited on
Commit
d0d8f33
·
verified ·
1 Parent(s): dc1ea84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def list_model_downloads(task: str)-> str:
27
  task: The name of a task for which huggingface models can be used.
28
  """
29
  most_downloaded_model = next(iter(list_models(filter=task, sort="downloads", direction=-1)))
30
- return "DistilBERT"#most_downloaded_model.id
31
 
32
 
33
  @tool
 
27
  task: The name of a task for which huggingface models can be used.
28
  """
29
  most_downloaded_model = next(iter(list_models(filter=task, sort="downloads", direction=-1)))
30
+ return most_downloaded_model.modelId
31
 
32
 
33
  @tool