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

Fix column order on refresh

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -231,7 +231,11 @@ def get_mteb_data(tasks=["Clustering"], langs=[], datasets=[], fillna=True, add_
231
  df = df.groupby("Model", as_index=False).first()
232
  # Put 'Model' column first
233
  cols = sorted(list(df.columns))
234
- cols.insert(0, cols.pop(cols.index("Model")))
 
 
 
 
235
  df = df[cols]
236
  if rank:
237
  df = add_rank(df)
 
231
  df = df.groupby("Model", as_index=False).first()
232
  # Put 'Model' column first
233
  cols = sorted(list(df.columns))
234
+ i = 0
235
+ for column in ["Model", "Model Size (Million Parameters)", "Memory Usage (GB, fp32)", "Embedding Dimensions", "Max Tokens"]:
236
+ if column in cols:
237
+ cols.insert(i, cols.pop(cols.index(column)))
238
+ i += 1
239
  df = df[cols]
240
  if rank:
241
  df = add_rank(df)