Upload app.py

app.py

@@ -53,7 +53,7 @@ def color_model_type_column(df, color_map):
     format_dict["Cost"] = "{:.2f}"
     
 
-    return df.style.applymap(apply_color, subset=["Category"]).format(format_dict, na_rep="")
+    return df.style.map(apply_color, subset=["Category"]).format(format_dict, na_rep="")
 
 
 def regex_table(dataframe, regex, filter_button, style=True):
@@ -147,7 +147,7 @@ with gr.Blocks(theme=theme) as app:
                             datatype=col_types_response_generation,
                             headers=df_response_generation.columns.tolist(),
                             elem_id="response_generation_dataframe",
-                            height=800,  # 800 px ≈ ~25 rows on default row-height
+                            row_count=(25, "dynamic"),
                         )
 
         with gr.TabItem("LLM Judge"):
@@ -189,7 +189,7 @@ with gr.Blocks(theme=theme) as app:
                             datatype=col_types_llm_judge,
                             headers=df_llm_judge.columns.tolist(),
                             elem_id="llm_judge_dataframe",
-                            height=800,  # 800 px ≈ ~25 rows on default row-height
+                            row_count=(25, "dynamic"),
                         )
             
     search_1.change(regex_table, inputs=[rewardbench_table_hidden, search_1, model_types_1], outputs=rewardbench_table)