Update app.py

app.py

@@ -61,19 +61,21 @@ with gr.Blocks(css=custom_css) as demo:
 
     with gr.Row() as udiff:
         with gr.Row():
-            drop = gr.Dropdown(["Object-Church", "Object-Parachute", "Object-Garbage","Style-Van Gogh"
-                               "Concept-Nudity", "Concept-Violence", "Illegal Activity"], 
+            drop = gr.Dropdown(["Object-Church", "Object-Parachute", "Object-Garbage","Style-Van Gogh",
+                               "Concept-Nudity", "Concept-Violence", "Illegal Activity", "None"], 
                                label="Unlearning undesirable")
         with gr.Column():
             # gr.Markdown("Please upload your model id.")
-            diffusion_model_id = gr.Textbox(label='diffusion_model_id')
-            concept = gr.Textbox(label='concept')
-            attacker = gr.Textbox(label='attacker')
+            drop_model = gr.Dropdown(["ESD", "FMN", "AC","UCE", "SLD"], 
+                               label="Unlearned DM")
+            # diffusion_model_T = gr.Textbox(label='diffusion_model_id')
+            # concept = gr.Textbox(label='concept')
+            # attacker = gr.Textbox(label='attacker')
 
-            start_button = gr.Button("Attack!")
+            # start_button = gr.Button("Attack!")
 
-        with gr.Column():
-            result = gr.Textbox(label="unsafe prompt")
+        # with gr.Column():
+        #     result = gr.Textbox(label="unsafe prompt")
 
     with gr.Column():
         gr.Examples(examples=[