merve HF Staff commited on
Commit
8ac54c7
·
1 Parent(s): 299a6bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,9 +22,9 @@ def make_clickable_model(model_name, repo_type, link=None):
22
 
23
  def get_repo_ids(repo_type):
24
  api = HfApi()
25
- if repo_type is "Model":
26
  notebooks = api.list_models(filter=["notebook-favorites"])
27
- elif repo_type is "Dataset":
28
  notebooks = api.list_datasets(filter=["notebook-favorites"])
29
  print(notebooks)
30
  notebook_ids = [x for x in notebooks]
 
22
 
23
  def get_repo_ids(repo_type):
24
  api = HfApi()
25
+ if repo_type == "Model":
26
  notebooks = api.list_models(filter=["notebook-favorites"])
27
+ elif repo_type == "Dataset":
28
  notebooks = api.list_datasets(filter=["notebook-favorites"])
29
  print(notebooks)
30
  notebook_ids = [x for x in notebooks]