eduagarcia commited on
Commit
349b10b
1 Parent(s): bbfe97c

Fix a weird bug that made the cicklabe model name fail to render in some boards

Browse files
Files changed (1) hide show
  1. app.py +3 -7
app.py CHANGED
@@ -221,7 +221,7 @@ def get_mteb_data(tasks=["Clustering"], langs=[], datasets=[], fillna=True, add_
221
  MODEL_EMB_DIM[model.modelId] = get_dim_seq_size(model)
222
  out["Embedding Dimensions"], out["Max Tokens"], out["Model Size (Million Parameters)"], out["Memory Usage (GB, fp32)"] = MODEL_EMB_DIM[model.modelId]
223
  except:
224
- MODEL_EMB_DIM[model.modelId] = None, None, None, None
225
  df_list.append(out)
226
  if model.library_name == "sentence-transformers" or "sentence-transformers" in model.tags or "modules.json" in {file.rfilename for file in model.siblings}:
227
  SENTENCE_TRANSFORMERS_COMPATIBLE_MODELS.add(out["Model"])
@@ -294,11 +294,7 @@ for board, board_config in BOARDS_CONFIG.items():
294
  all_data_tasks.extend(data_tasks.values())
295
  else:
296
  for task_category, task_category_list in board_config["tasks"].items():
297
- columns = ["Model", "Model Size (Million Parameters)", "Memory Usage (GB, fp32)", "Embedding Dimensions", "Max Tokens"]
298
- if len(task_category_list) > 1:
299
- columns.append("Average")
300
- print(board, task_category, task_category_list)
301
- data_task_category = get_mteb_data([task_category], [], task_category_list, refresh=False)[columns + task_category_list]
302
  boards_data[board]["data_tasks"][task_category] = data_task_category
303
  all_data_tasks.append(data_task_category)
304
 
@@ -388,7 +384,7 @@ for board, board_config in BOARDS_CONFIG.items():
388
  "language_long": board_config["language_long"],
389
  "description": f"**{task_category} {board_pretty_name}** {task_icon}{board_icon}",
390
  "data": boards_data[board]["data_tasks"][task_category],
391
- "refresh": partial(get_mteb_data, [task_category], [], task_category_list),
392
  "credits": credits,
393
  })
394
 
 
221
  MODEL_EMB_DIM[model.modelId] = get_dim_seq_size(model)
222
  out["Embedding Dimensions"], out["Max Tokens"], out["Model Size (Million Parameters)"], out["Memory Usage (GB, fp32)"] = MODEL_EMB_DIM[model.modelId]
223
  except:
224
+ MODEL_EMB_DIM[model.modelId] = "", "", "", ""
225
  df_list.append(out)
226
  if model.library_name == "sentence-transformers" or "sentence-transformers" in model.tags or "modules.json" in {file.rfilename for file in model.siblings}:
227
  SENTENCE_TRANSFORMERS_COMPATIBLE_MODELS.add(out["Model"])
 
294
  all_data_tasks.extend(data_tasks.values())
295
  else:
296
  for task_category, task_category_list in board_config["tasks"].items():
297
+ data_task_category = get_mteb_data(tasks=[task_category], datasets=task_category_list, refresh=False)
 
 
 
 
298
  boards_data[board]["data_tasks"][task_category] = data_task_category
299
  all_data_tasks.append(data_task_category)
300
 
 
384
  "language_long": board_config["language_long"],
385
  "description": f"**{task_category} {board_pretty_name}** {task_icon}{board_icon}",
386
  "data": boards_data[board]["data_tasks"][task_category],
387
+ "refresh": partial(get_mteb_data, tasks=[task_category], datasets=task_category_list),
388
  "credits": credits,
389
  })
390