Update app.py
Browse files
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
|
26 |
notebooks = api.list_models(filter=["notebook-favorites"])
|
27 |
-
elif repo_type
|
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]
|