Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
eduagarcia
commited on
Commit
•
a20529c
1
Parent(s):
349b10b
Fix column order on refresh
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
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)
|