Update app.py

app.py

@@ -107,15 +107,15 @@ def update_metrics(selected_benchmarks):
 def update_leaderboard(selected_methods, selected_metrics):
     df = get_baseline_df(selected_methods, selected_metrics)
     df = df.applymap(lambda x: round(x, 4) if isinstance(x, (int, float)) else x)
-    styled = df.style.applymap(style_methods, subset=["Method"])
-    html = styled.set_table_attributes('class="dataframe"').render()
-    return html
+    df['Method'] = df['Method'].apply(format_method)
+    return df
 
-def style_methods(val):
-    """Given a model name, return a CSS style string."""
-    c = color_dict.get(val, "black")
-    return f"color: {c}"
 
+def format_method(name: str) -> str:
+    """Wrap a model name in a coloured <span> so Gradio renders it."""
+    colour = color_dict.get(name, "black")
+    return f"<span style='color:{colour}; font-weight:bold;'>{name}</span>"
+    
 # ------- Visualisation helpers ---------------------------------------------
 
 
@@ -222,6 +222,7 @@ with block:
 
             baseline_value = get_baseline_df(method_names, metric_names)
             baseline_value = baseline_value.applymap(lambda x: round(x, 4) if isinstance(x, (int, float)) else x)
+            baseline_value['Method'] = baseline_value['Method'].apply(format_method)
             baseline_header  = ["Method"] + metric_names
             baseline_datatype = ['markdown'] + ['number'] * len(metric_names)
 
@@ -233,7 +234,7 @@ with block:
             
             with gr.Row(show_progress=True, variant='panel'):
                 data_component = gr.Dataframe(
-                    value=initial_html,
+                    value=baseline_value,
                     headers=baseline_header,
                     type="pandas",
                     datatype=baseline_datatype,