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','illgel_activity']
+files = ['church','garbage','parachute','tench', 'vangogh', 'nidity', 'violence','illegal_activity']
 df_results_init = df_results.copy()[show_columns]
 
 def update_table(
@@ -227,7 +227,26 @@ with demo:
                     #         interactive=True,)
             for i in range(len(files)):
                 if files[i] == "church":
-                    name = "### Unlearned Concepts "+" Church" 
+                    name = "### Unlearned Objects "+" Church" 
+                elif file[i] == 'garbage':
+                    name = "### Unlearned Objects "+" Garbage"
+                elif file[i] == 'tench':
+                    name = "### Unlearned Objects "+" Tench"
+                elif file[i] == 'parachute':
+                    name = "### Unlearned Objects "+" Parachute"
+                elif file[i] == 'vangogh':
+                    name = "### Unlearned Stype "+" Van Gogh"
+                elif file[i] == 'nudity':
+                    name = "### Unlearned  Concepts "+" Nudity"
+                elif file[i] == 'violence':
+                    name = "### Unlearned Concepts "+" Violence"
+                elif file[i] == 'ill':
+                    name = "### Unlearned Concepts "+" Illgal Activity"
+                
+                
+                
+                
+                
                 gr.Markdown(name)
             gr.Markdown("### Unlearned Concepts Parachute")
             leaderboard_table = gr.components.Dataframe(