[Update]Change read initial files

app.py

@@ -75,13 +75,14 @@ def restart_space():
 #     running_eval_queue_df,
 #     pending_eval_queue_df,
 # ) = get_evaluation_queue_df(EVAL_REQUESTS_PATH, EVAL_COLS)
-csv_path='./assets/parachute.csv'
-df_results = load_data(csv_path)
-methods = list(set(df_results['Unlearned_Methods']))
+
 all_columns = ['Unlearned_Methods','Source', 'Diffusion_Models','Pre-ASR','Pre-ASR','Post-FID']
 show_columns = ['Unlearned_Methods','Source', 'Diffusion_Models','Pre-ASR','Pre-ASR','Post-FID']
 TYPES = ['str', 'markdown', 'str', 'number', 'number', 'number', 'number']
 files = ['church','garbage','parachute','tench', 'vangogh', 'nudity', 'violence','illegal_activity']
+csv_path='./assets/'+files[0]+'cvs'
+df_results = load_data(csv_path)
+methods = list(set(df_results['Unlearned_Methods']))
 df_results_init = df_results.copy()[show_columns]
 
 def update_table(
@@ -229,32 +230,29 @@ with demo:
                 if files[0] == "church":
                     name = "### Unlearned Objects "+" Church" 
                     csv_path = './assets/'+files[i]+'.csv'
-                # elif files[i] == 'garbage':
-                #     name = "### Unlearned Objects "+" Garbage"
-                #     csv_path = './assets/'+files[i]+'.csv'
-                # elif files[i] == 'tench':
-                #     name = "### Unlearned Objects "+" Tench"
-                #     csv_path = './assets/'+files[i]+'.csv'
-                # elif files[i] == 'parachute':
-                #     name = "### Unlearned Objects "+" Parachute"
-                #     csv_path = './assets/'+files[i]+'.csv'
-                # elif files[i] == 'vangogh':
-                #     name = "### Unlearned Stype "+" Van Gogh"
-                #     csv_path = './assets/'+files[i]+'.csv'
-                # elif files[i] == 'nudity':
-                #     name = "### Unlearned  Concepts "+" Nudity"
-                #     csv_path = './assets/'+files[i]+'.csv'
-                # elif files[i] == 'violence':
-                #     name = "### Unlearned Concepts "+" Violence"
-                #     csv_path = './assets/'+files[i]+'.csv'
-                # elif files[i] == 'illegal_activity':
-                #     name = "### Unlearned Concepts "+" Illgal Activity"
-                #     csv_path = './assets/'+files[i]+'.csv'
-                
-                
-                
+                elif files[i] == 'garbage':
+                    name = "### Unlearned Objects "+" Garbage"
+                    csv_path = './assets/'+files[i]+'.csv'
+                elif files[i] == 'tench':
+                    name = "### Unlearned Objects "+" Tench"
+                    csv_path = './assets/'+files[i]+'.csv'
+                elif files[i] == 'parachute':
+                    name = "### Unlearned Objects "+" Parachute"
+                    csv_path = './assets/'+files[i]+'.csv'
+                elif files[i] == 'vangogh':
+                    name = "### Unlearned Stype "+" Van Gogh"
+                    csv_path = './assets/'+files[i]+'.csv'
+                elif files[i] == 'nudity':
+                    name = "### Unlearned  Concepts "+" Nudity"
+                    csv_path = './assets/'+files[i]+'.csv'
+                elif files[i] == 'violence':
+                    name = "### Unlearned Concepts "+" Violence"
+                    csv_path = './assets/'+files[i]+'.csv'
+                elif files[i] == 'illegal_activity':
+                    name = "### Unlearned Concepts "+" Illgal Activity"
+                    csv_path = './assets/'+files[i]+'.csv'
                 
-                print(i)
+            
                 gr.Markdown(name)
                 df_results = load_data(csv_path)
                 df_results_init = df_results.copy()[show_columns]