[Update]change FID to pre-FID and post-FID

app.py

@@ -34,8 +34,8 @@ from dummydatagen import dummy_data_for_plot, create_metric_plot_obj_1, dummydf
 import copy
 
 def load_data(data_path):
-    columns = ['Unlearned_Methods','Source', 'Diffusion_Models','Pre-ASR', 'Post-ASR','FID']
-    columns_sorted = ['Unlearned_Methods','Source', 'Diffusion_Models','Pre-ASR', 'Post-ASR','FID']
+    columns = ['Unlearned_Methods','Source', 'Diffusion_Models','Pre-ASR', 'Post-ASR','Pre-FID', 'Post-FID']
+    columns_sorted = ['Unlearned_Methods','Source', 'Diffusion_Models','Pre-ASR', 'Post-ASR','Pre-FID', 'Post-FID']
     
     df = pd.read_csv(data_path).dropna()
     df['Post-ASR'] = df['Post-ASR'].round(0)
@@ -78,8 +78,8 @@ def restart_space():
 csv_path='./assets/object_parachute.csv'
 df_results = load_data(csv_path)
 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']
+all_columns = ['Unlearned_Methods','Source', 'Diffusion_Models','Pre-ASR', 'Post-ASR','Pre-FID']
+show_columns = ['Unlearned_Methods','Source', 'Diffusion_Models','Pre-ASR', 'Post-ASR','Post-FID']
 TYPES = ['str', 'markdown', 'str', 'number', 'number', 'number']
 files = ['church','garbage','parachute','tench', 'vangogh', 'nudity', 'violence','illegal_activity']
 df_results_init = df_results.copy()[show_columns]
@@ -157,7 +157,7 @@ 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']
+    all_columns =['Pre-ASR','Post-ASR','PreFID','Post-FID']
 
     if (len(columns_1)) == 0:
         filtered_df = df[
@@ -193,7 +193,7 @@ with demo:
                     with gr.Row():
                         model1_column = gr.CheckboxGroup(
                         label="Evaluation Metrics",
-                        choices=['Pre-ASR', 'Post-ASR','FID'],
+                        choices=['Pre-ASR', 'Post-ASR','Pre-FID','Post-FID'],
                         interactive=True,
                         elem_id="column-select",
                     )