Update app.py

app.py

@@ -107,6 +107,11 @@ def update_metrics(selected_benchmarks):
 def update_leaderboard(selected_methods, selected_metrics):
     return get_baseline_df(selected_methods, selected_metrics)
 
+def style_methods(val):
+    """Given a model name, return a CSS style string."""
+    c = color_dict.get(val, "black")
+    return f"color: {c}"
+
 # ------- Visualisation helpers ---------------------------------------------
 
 
@@ -216,9 +221,14 @@ with block:
             baseline_header  = ["Method"] + metric_names
             baseline_datatype = ['markdown'] + ['number'] * len(metric_names)
 
+            baseline_styler = baseline_value.style.applymap(
+                style_methods,
+                subset=["Method"]
+            )
+            
             with gr.Row(show_progress=True, variant='panel'):
                 data_component = gr.Dataframe(
-                    value=baseline_value,
+                    value=baseline_styler,
                     headers=baseline_header,
                     type="pandas",
                     datatype=baseline_datatype,
@@ -226,6 +236,7 @@ with block:
                     elem_id="leaderboard-table",
                     pinned_columns=1,
                     max_height=1000,
+                    show_fullscreen_button=True
                 )
 
             # callbacks