Update app.py

app.py

@@ -81,7 +81,7 @@ methods = list(set(df_results['Unlearned_Methods']))
 all_columns = ['Unlearned_Methods','Source', 'Diffusion_Models','Pre-ASR', 'Post-ASR','FID']
 show_columns = ['Unlearned_Methods','Source', 'Diffusion_Models','Pre-ASR', 'Post-ASR','FID']
 TYPES = ['str', 'markdown', 'str', 'number', 'number', 'number']
-files = ['church','garbage','parachute','tench', 'vangogh', 'nidity', 'violence','illegal_activity']
+files = ['church','garbage','parachute','tench', 'vangogh', 'nudity', 'violence','illegal_activity']
 df_results_init = df_results.copy()[show_columns]
 
 def update_table(
@@ -247,7 +247,7 @@ with demo:
                 elif files[i] == 'violence':
                     name = "### Unlearned Concepts "+" Violence"
                     csv_path = './assets/'+files[i]+'.csv'
-                elif files[i] == 'ill':
+                elif files[i] == 'illegal_activity':
                     name = "### Unlearned Concepts "+" Illgal Activity"
                     csv_path = './assets/'+files[i]+'.csv'
                 
@@ -283,6 +283,17 @@ with demo:
                     ],
                     leaderboard_table,
                 )
+
+                for selector in [model1_column]:
+                selector.change(
+                    update_table,
+                    [   
+                        hidden_leaderboard_table_for_search,
+                        model1_column,
+                        search_bar,
+                    ],
+                    leaderboard_table,
+                )