add progress bar

app.py

@@ -7,7 +7,7 @@ from huggingface_hub import get_collection
 from cytoolz import groupby
 from collections import defaultdict
 import os
-
+from tqdm.auto import tqdm
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 is_macos = platform.system() == "Darwin"
 LIMIT = None
@@ -15,12 +15,12 @@ CACHE_TIME = 60 * 5 # 5 minutes
 
 @cached(cache=TTLCache(maxsize=100, ttl=CACHE_TIME))
 def get_models():
-    return list(iter(list_models(full=True, limit=LIMIT)))
+    return list(tqdm(iter(list_models(full=True, limit=LIMIT))))
 
 
 @cached(cache=TTLCache(maxsize=100, ttl=CACHE_TIME))
 def get_datasets():
-    return list(iter(list_datasets(full=True, limit=LIMIT)))
+    return list(tqdm(iter(list_datasets(full=True, limit=LIMIT))))
 
 
 get_models()  # warm up the cache