update rank feature

#19
by MINGYISU - opened
Files changed (1) hide show
  1. utils.py +3 -2
utils.py CHANGED
@@ -13,14 +13,14 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
13
  TASKS = ["Classification", "VQA", "Retrieval", "Grounding"]
14
 
15
  MODEL_INFO = [
16
- "Models", "Model Size(B)", "Data Source",
17
  "Overall",
18
  "Classification", "VQA", "Retrieval", "Grounding"
19
  ]
20
 
21
  BASE_COLS = [col for col in MODEL_INFO if col not in TASKS]
22
 
23
- DATA_TITLE_TYPE = ['markdown', 'str', 'markdown', 'number', 'number', 'number', 'number', 'number', 'number', 'number']
24
 
25
  SUBMISSION_NAME = "MMEB"
26
  SUBMISSION_URL = os.path.join("https://huggingface.co/spaces/TIGER-Lab/", SUBMISSION_NAME)
@@ -142,6 +142,7 @@ def get_df():
142
  df['Model Size(B)'] = df['Model Size(B)'].apply(process_model_size)
143
  df = df.sort_values(by=['Overall'], ascending=False)
144
  df = create_hyperlinked_names(df)
 
145
  return df
146
 
147
 
 
13
  TASKS = ["Classification", "VQA", "Retrieval", "Grounding"]
14
 
15
  MODEL_INFO = [
16
+ "Rank", "Models", "Model Size(B)", "Data Source",
17
  "Overall",
18
  "Classification", "VQA", "Retrieval", "Grounding"
19
  ]
20
 
21
  BASE_COLS = [col for col in MODEL_INFO if col not in TASKS]
22
 
23
+ DATA_TITLE_TYPE = ['number', 'markdown', 'str', 'markdown', 'number', 'number', 'number', 'number', 'number', 'number', 'number']
24
 
25
  SUBMISSION_NAME = "MMEB"
26
  SUBMISSION_URL = os.path.join("https://huggingface.co/spaces/TIGER-Lab/", SUBMISSION_NAME)
 
142
  df['Model Size(B)'] = df['Model Size(B)'].apply(process_model_size)
143
  df = df.sort_values(by=['Overall'], ascending=False)
144
  df = create_hyperlinked_names(df)
145
+ df['Rank'] = range(1, len(df) + 1)
146
  return df
147
 
148