update space

app.py

@@ -76,20 +76,20 @@ def init_leaderboard(dataframe):
         ),
         search_columns=[AutoEvalColumn.model.name, AutoEvalColumn.license.name],
         hide_columns=[c.name for c in fields(AutoEvalColumn) if c.hidden],
-        # filter_columns=[
-        #     ColumnFilter(AutoEvalColumn.model_type.name, type="checkboxgroup", label="Model types"),
-        #     ColumnFilter(AutoEvalColumn.precision.name, type="checkboxgroup", label="Precision"),
-        #     ColumnFilter(
-        #         AutoEvalColumn.params.name,
-        #         type="slider",
-        #         min=0.01,
-        #         max=150,
-        #         label="Select the number of parameters (B)",
-        #     ),
-        #     ColumnFilter(
-        #         AutoEvalColumn.still_on_hub.name, type="boolean", label="Deleted/incomplete", default=True
-        #     ),
-        # ],
+        filter_columns=[
+            ColumnFilter(AutoEvalColumn.model_type.name, type="checkboxgroup", label="Model types"),
+            ColumnFilter(AutoEvalColumn.precision.name, type="checkboxgroup", label="Precision"),
+            ColumnFilter(
+                AutoEvalColumn.params.name,
+                type="slider",
+                min=0.01,
+                max=150,
+                label="Select the number of parameters (B)",
+            ),
+            ColumnFilter(
+                AutoEvalColumn.still_on_hub.name, type="boolean", label="Deleted/incomplete", default=False
+            ),
+        ],
         bool_checkboxgroup_label="Hide models",
         interactive=False,
     )

src/display/utils.py

@@ -65,6 +65,7 @@ class ModelType(Enum):
     PT = ModelDetails(name="pretrained", symbol="🟢")
     FT = ModelDetails(name="fine-tuned", symbol="🔶")
     IFT = ModelDetails(name="instruction-tuned", symbol="⭕")
+    CT = ModelDetails(name="chat-tuned", symbol="💬")
     RL = ModelDetails(name="RL-tuned", symbol="🟦")
     Unknown = ModelDetails(name="", symbol="?")
 
@@ -81,6 +82,8 @@ class ModelType(Enum):
             return ModelType.RL
         if "instruction-tuned" in type or "⭕" in type:
             return ModelType.IFT
+        if "chat-tuned" in type or "💬" in type:
+            return ModelType.CT
         return ModelType.Unknown
 
 class WeightType(Enum):
@@ -103,6 +106,7 @@ class Precision(Enum):
 # Column selection
 COLS = [c.name for c in fields(AutoEvalColumn) if not c.hidden]
 
+
 EVAL_COLS = [c.name for c in fields(EvalQueueColumn)]
 EVAL_TYPES = [c.type for c in fields(EvalQueueColumn)]