Update app.py

app.py

@@ -226,30 +226,30 @@ with demo:
                     #         step=1, label="Attacking Steps", info="Choose between 1 and 100",
                     #         interactive=True,)
             for i in range(len(files)):
-                if files[i] == "church":
+                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'