Update app.py

app.py

@@ -168,7 +168,7 @@ def filter_models(
 
     # モデルサイズでフィルタリング
     if "Unknown" in size_query:
-        size_mask = filtered_df[AutoEvalColumn.params.name].isna() | (filtered_df[AutoEvalColumn.params.name] == "Unknown")
+        size_mask = filtered_df[AutoEvalColumn.params.name].isna() | (filtered_df[AutoEvalColumn.params.name].astype(str).isin(["Unknown", "?"]))
     else:
         numeric_interval = pd.IntervalIndex(sorted([NUMERIC_INTERVALS[s] for s in size_query if s != "Unknown"]))
         params_column = pd.to_numeric(filtered_df[AutoEvalColumn.params.name], errors="coerce")
@@ -266,10 +266,10 @@ with demo:
             leaderboard_df_filtered = 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)
 
             # 列名の重複を解消
-            leaderboard_df_filtered.columns = pd.io.parsers.readers._maybe_dedup_names(leaderboard_df_filtered.columns)
+            leaderboard_df_filtered.columns = pd.io.parsers.base._maybe_dedup_names(leaderboard_df_filtered.columns)
             
             leaderboard_table = gr.components.Dataframe(
-                value=leaderboard_df_filtered,
+                value=leaderboard_df_filtered.to_dict('records'),
                 headers=list(leaderboard_df_filtered.columns),
                 datatype={col: "str" for col in leaderboard_df_filtered.columns},
                 elem_id="leaderboard-table",
@@ -277,8 +277,8 @@ with demo:
                 visible=True,
             )
             print("Leaderboard table initial value:")
-            print(leaderboard_table.value)
-            print(f"Leaderboard table shape: {leaderboard_table.value.shape if isinstance(leaderboard_table.value, pd.DataFrame) else len(leaderboard_table.value)}")
+            print(leaderboard_df_filtered)
+            print(f"Leaderboard table shape: {leaderboard_df_filtered.shape}")
             
             # Dummy leaderboard for handling the case when the user uses backspace key
             hidden_leaderboard_table_for_search = gr.components.Dataframe(