davanstrien HF staff commited on
Commit
1e378d7
1 Parent(s): d7a9899

add total children downloads

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -8,7 +8,7 @@ from cachetools import cached, TTLCache
8
 
9
  @cached(TTLCache(maxsize=10, ttl=60 * 60 * 3))
10
  def get_all_models():
11
- models = list(tqdm(iter(list_models(cardData=True))))
12
  return [model for model in models if model is not None]
13
 
14
 
@@ -49,7 +49,7 @@ unique_pipeline_tags = list(
49
  def parse_org(hub_id):
50
  parts = hub_id.split("/")
51
  if len(parts) == 2:
52
- return parts[0] if parts[0] != '.' else None
53
  else:
54
  return "huggingface"
55
 
@@ -69,8 +69,13 @@ def return_models_for_base_model(base_model):
69
  # sort models by downloads
70
  models = sorted(models, key=lambda x: x.downloads, reverse=True)
71
  results = ""
72
- results += f"## {base_model} children\n\n"
73
  results += f"{base_model} has {len(models)} children\n\n"
 
 
 
 
 
74
  for model in models:
75
  url = f"https://huggingface.co/{model.modelId}"
76
  results += (
 
8
 
9
  @cached(TTLCache(maxsize=10, ttl=60 * 60 * 3))
10
  def get_all_models():
11
+ models = list(tqdm(iter(list_models(cardData=True, limit=None))))
12
  return [model for model in models if model is not None]
13
 
14
 
 
49
  def parse_org(hub_id):
50
  parts = hub_id.split("/")
51
  if len(parts) == 2:
52
+ return parts[0] if parts[0] != "." else None
53
  else:
54
  return "huggingface"
55
 
 
69
  # sort models by downloads
70
  models = sorted(models, key=lambda x: x.downloads, reverse=True)
71
  results = ""
72
+ results += f"## [`{base_model}`](https://huggingface.co/{base_model}) children\n\n"
73
  results += f"{base_model} has {len(models)} children\n\n"
74
+ total_download_number = sum(model.downloads for model in models)
75
+ results += (
76
+ f"`{base_model}`'s children have been"
77
+ f" downloaded {total_download_number:,} times\n\n"
78
+ )
79
  for model in models:
80
  url = f"https://huggingface.co/{model.modelId}"
81
  results += (