Update app.py

app.py

@@ -105,13 +105,11 @@ def update_metrics(selected_benchmarks):
 
 
 def update_leaderboard(selected_methods, selected_metrics):
-    # 1) get the df
     df = get_baseline_df(selected_methods, selected_metrics)
-    # 2) round floats
     df = df.applymap(lambda x: round(x, 4) if isinstance(x, (int, float)) else x)
-    # 3) apply your colour‐styling to the Method column
     styled = df.style.applymap(style_methods, subset=["Method"])
-    return styled
+    html = styled.set_table_attributes('class="dataframe"').render()
+    return html
 
 def style_methods(val):
     """Given a model name, return a CSS style string."""
@@ -227,14 +225,15 @@ with block:
             baseline_header  = ["Method"] + metric_names
             baseline_datatype = ['markdown'] + ['number'] * len(metric_names)
 
-            baseline_styler = baseline_value.style.applymap(
-                style_methods,
-                subset=["Method"]
+            initial_html = baseline_value.style\
+                .applymap(style_methods, subset=["Method"])\
+                .set_table_attributes('class="dataframe"')\
+                .render()
             )
             
             with gr.Row(show_progress=True, variant='panel'):
                 data_component = gr.Dataframe(
-                    value=baseline_styler,
+                    value=initial_html,
                     headers=baseline_header,
                     type="pandas",
                     datatype=baseline_datatype,