refine

app.py

@@ -184,6 +184,7 @@ with demo:
 
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
         with gr.TabItem("UnlearnDiffAtk Benchmark", elem_id="UnlearnDiffAtk-benchmark-tab-table", id=0):
+            files = ['nudity','vangogh', 'church','garbage','parachute','tench']
             with gr.Row():
                 with gr.Column():
                     with gr.Row():
@@ -269,6 +270,7 @@ with demo:
                 )
                     
         with gr.TabItem("Style", elem_id="Style", id=1):
+            files = ['vangogh']
             with gr.Row():
                 with gr.Column():
                     with gr.Row():