[Update]add model1_column

app.py

@@ -86,7 +86,7 @@ df_results_init = df_results.copy()[show_columns]
 
 def update_table(
     hidden_df: pd.DataFrame,
-    # columns: list,
+    model1_column: list,
     #type_query: list,
     open_query: list,
     # precision_query: str,
@@ -102,7 +102,7 @@ def update_table(
     # filtered_df = filtered_df[filtered_df['Unlearned_Methods'].isin(open_query)]
     # map_open = {'open': 'Yes', 'closed': 'No'}
     # filtered_df = filtered_df[filtered_df['Open?'].isin([map_open[o] for o in open_query])]
-
+    filtered_df=select_columns(filtered_df,open_query)
     filtered_df = filter_queries(query, filtered_df)
     # filtered_df = filtered_df[[map_columns[k] for k in columns]]
     # deduplication
@@ -131,6 +131,27 @@ def filter_queries(query: str, filtered_df: pd.DataFrame) -> pd.DataFrame:
 
     return filtered_df
 
+def select_columns(df: pd.DataFrame, columns_1: list) -> pd.DataFrame:
+    always_here_cols = ['Unlearned_Methods','Source', 'Diffusion_Models']
+    
+    # We use COLS to maintain sorting
+    all_columns =['Pre-ASR','Post-ASR','FID']
+
+    if (len(columns_1)) == 0:
+        filtered_df = df[
+            always_here_cols +
+            [c for c in all_columns if c in df.columns]
+        ]
+
+    else:
+        filtered_df = df[
+            always_here_cols +
+            [c for c in all_columns if c in df.columns and (c in columns_1) ]
+        ]
+
+    return filtered_df
+
+
 demo = gr.Blocks(css=custom_css)
 with demo:
     gr.HTML(TITLE)
@@ -212,6 +233,7 @@ with demo:
                 [   
                     # df_avg,
                     hidden_leaderboard_table_for_search,
+                    model1_column,
                     # shown_columns,
                     #type_query,
                     open_query,
@@ -225,12 +247,13 @@ with demo:
             )
 
             #for selector in [type_query, open_query]:
-            for selector in [open_query]:
+            for selector in [open_query,model1_column]:
                 selector.change(
                     update_table,
                     [   
                         # df_avg,
                         hidden_leaderboard_table_for_search,
+                        model1_column,
                         # shown_columns,
                         #type_query,
                         open_query,