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

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 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 "DistilBERT"#most_downloaded_model.id
31
 
32
 
33
  @tool