Update app.py

app.py

@@ -146,9 +146,6 @@ def filter_models(
     filtered_df = filtered_df.loc[df[AutoEvalColumn.precision.name].isin(precision_query + ["None"])]
     filtered_df = filtered_df.loc[df[AutoEvalColumn.add_special_tokens.name].isin(add_special_tokens_query)]
     filtered_df = filtered_df.loc[df[AutoEvalColumn.num_few_shots.name].isin(num_few_shots_query)]
-    print("Filtered DataFrame shape:", filtered_df.shape)
-    print("Filtered DataFrame columns:", filtered_df.columns.tolist())
-    print("Filtered DataFrame sample:", filtered_df.head().to_dict('records'))
     
 
     numeric_interval = pd.IntervalIndex(sorted([NUMERIC_INTERVALS[s] for s in size_query]))
@@ -157,12 +154,7 @@ def filter_models(
     filtered_df = filtered_df.loc[mask]
     return filtered_df
 
-filtered_df = filter_models(leaderboard_df, [t.to_str(" : ") for t in ModelType], list(NUMERIC_INTERVALS.keys()), [i.value.name for i in Precision], [i.value.name for i in AddSpecialTokens], [i.value.name for i in NumFewShots], False, False, False)
-display_columns = [c.name for c in fields(AutoEvalColumn) if c.never_hidden] + shown_columns.value
-display_data = filtered_df[display_columns].to_dict('records')
-
-print("Display columns:", display_columns)
-print("Display data sample:", display_data[:1])
+leaderboard_df = filter_models(leaderboard_df, [t.to_str(" : ") for t in ModelType], list(NUMERIC_INTERVALS.keys()), [i.value.name for i in Precision], [i.value.name for i in AddSpecialTokens], [i.value.name for i in NumFewShots], False, False, False)
 
 demo = gr.Blocks(css=custom_css)
 with demo:
@@ -244,12 +236,17 @@ with demo:
                     )
 
             leaderboard_table = gr.components.Dataframe(
-                value=display_data,
-                headers=display_columns,
-                datatype={col: str(TYPES.get(col, "str")) for col in display_columns},
+                value=leaderboard_df[
+                    [c.name for c in fields(AutoEvalColumn) if c.never_hidden]
+                    + shown_columns.value
+                    # + [AutoEvalColumn.dummy.name]
+                ],
+                headers=[c.name for c in fields(AutoEvalColumn) if c.never_hidden] + shown_columns.value,
+                datatype=TYPES,
                 elem_id="leaderboard-table",
                 interactive=False,
                 visible=True,
+                #column_widths=["2%", "33%"] 
             )
             print(leaderboard_df.head())  # リーダーボードテーブルに渡される前のデータを確認