Nathan Habib commited on
Commit
5da3025
1 Parent(s): bc44e71
src/display_models/get_model_metadata.py CHANGED
@@ -26,7 +26,6 @@ def get_model_infos_from_hub(leaderboard_data: List[dict]):
26
 
27
  for model_data in tqdm(leaderboard_data):
28
  model_name = model_data["model_name_for_query"]
29
- <<<<<<< HEAD
30
 
31
  if model_name in model_info_cache:
32
  model_info = model_info_cache[model_name]
@@ -40,16 +39,6 @@ def get_model_infos_from_hub(leaderboard_data: List[dict]):
40
  model_data[AutoEvalColumn.likes.name] = None
41
  model_data[AutoEvalColumn.params.name] = get_model_size(model_name, None)
42
  continue
43
- =======
44
- try:
45
- model_info = api.model_info(model_name)
46
- except huggingface_hub.utils._errors.RepositoryNotFoundError:
47
- print("Repo not found!", model_name)
48
- model_data[AutoEvalColumn.license.name] = "?"
49
- model_data[AutoEvalColumn.likes.name] = 0
50
- model_data[AutoEvalColumn.params.name] = get_model_size(model_name, None)
51
- continue
52
- >>>>>>> 6e79cea283b9033350b77806ca64c34a2e0cd323
53
 
54
  model_data[AutoEvalColumn.license.name] = get_model_license(model_info)
55
  model_data[AutoEvalColumn.likes.name] = get_model_likes(model_info)
 
26
 
27
  for model_data in tqdm(leaderboard_data):
28
  model_name = model_data["model_name_for_query"]
 
29
 
30
  if model_name in model_info_cache:
31
  model_info = model_info_cache[model_name]
 
39
  model_data[AutoEvalColumn.likes.name] = None
40
  model_data[AutoEvalColumn.params.name] = get_model_size(model_name, None)
41
  continue
 
 
 
 
 
 
 
 
 
 
42
 
43
  model_data[AutoEvalColumn.license.name] = get_model_license(model_info)
44
  model_data[AutoEvalColumn.likes.name] = get_model_likes(model_info)