Update app.py

app.py

@@ -796,19 +796,26 @@ for column in DATA_OVERALL.columns:
             if DATA_OVERALL[column].dtype == 'object':
                 DATA_OVERALL[column] = DATA_OVERALL[column].apply(remove_invalid_unicode)
 
+from dataclasses import dataclass
+
+@dataclass
+class LeaderboardColumn:
+    name: str
+    type: str 
+    
 DATA_OVERALL_COLUMN_TO_DATATYPE = [
-    "Rank", "number",
-    "Model Size (GB)", "number",
-    "Embedding Dimensions", "number",
-    "Max Tokens", "number",
-    "Average", "number",
-    "Classification Average", "number",
-    "Classification Average", "number",
-    "Pair Classification Average", "number",
-    "Reranking Average", "number",
-    "Retrieval Average", "number",
-    "STS Average", "number",
-    "Summarization Average", "number"
+    LeaderboardColumn("Rank", "number"),
+    LeaderboardColumn("Model Size (GB)", "number"),
+    LeaderboardColumn("Embedding Dimensions", "number"),
+    LeaderboardColumn("Max Tokens", "number"),
+    LeaderboardColumn("Average", "number"),
+    LeaderboardColumn("Classification Average", "number"),
+    LeaderboardColumn("Classification Average", "number"),
+    LeaderboardColumn("Pair Classification Average", "number"),
+    LeaderboardColumn("Reranking Average", "number"),
+    LeaderboardColumn("Retrieval Average", "number"),
+    LeaderboardColumn("STS Average", "number"),
+    LeaderboardColumn("Summarization Average", "number")
 ]
 
 COLS = [col.name for col in DATA_OVERALL_COLUMN_TO_DATATYPE]