Update app.py

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','FID']
+    columns_sorted = ['Unlearned_Methods','Source', 'Diffusion_Models','Pre-ASR', 'Post-ASR','FID']
     
     df = pd.read_csv(data_path, usecols=columns).dropna()
     df['Post-ASR'] = df['Post-ASR'].round(0)