xinchen9 commited on
Commit
1d193c7
·
verified ·
1 Parent(s): e7e8aa5

[Update]Adjust indent

Browse files
Files changed (1) hide show
  1. app.py +18 -17
app.py CHANGED
@@ -28,25 +28,26 @@ with gr.Blocks() as demo:
28
  drop = gr.Dropdown(["Object-Church", "Object-Parachute", "Object-Garbage_Truck","Object-Tench",
29
  "Style-Van_Gogh","Concept-Nudity"],
30
  label="Unlearning undesirable")
31
- with gr.Column():
32
- # gr.Markdown("Please upload your model id.")
33
- drop_model = gr.Dropdown(["Erased Stable Diffusion(ESD)", "Forget-me-not(FMN)", "Ablating concepts(AC)","Unified Concept Editing(UCE)", "(Safe Latent Diffusion)SLD"],
34
- label="Unlearned DMs")
35
- with gr.Column():
36
- # gr.Markdown("Please upload your model id.")
37
- drop_text = gr.Dropdown(["Object-Church", "Object-Parachute", "Object-Garbage_Truck","Object-Tench",
38
- "Style-Van_Gogh","Concept-Nudity", "None"],
39
- label="AdvUnlearn Text Encoder")
40
 
41
  with gr.Row() as attack:
42
  text_input = gr.Textbox(label="Prompt")
43
- with gr.Column(min_width=260):
44
-
45
- img1 = gr.Image("images/cheetah.jpg",label="Image Generated without AdvUnlearn",width=260,show_share_button=False,show_download_button=False)
46
- with gr.Column():
47
- start_button = gr.Button("AdvUnlearn",size='lg')
48
- with gr.Column(min_width=260):
49
- text_ouput = gr.Textbox(label="Prompt Genetated by UnlearnDiffAtk")
50
- img2 = gr.Image("images/cheetah.jpg",label="Image Generated without AdvUnlearn",width=260,show_share_button=False,show_download_button=False)
 
51
 
52
  demo.launch()
 
28
  drop = gr.Dropdown(["Object-Church", "Object-Parachute", "Object-Garbage_Truck","Object-Tench",
29
  "Style-Van_Gogh","Concept-Nudity"],
30
  label="Unlearning undesirable")
31
+ with gr.Column():
32
+ # gr.Markdown("Please upload your model id.")
33
+ drop_model = gr.Dropdown(["Erased Stable Diffusion(ESD)", "Forget-me-not(FMN)", "Ablating concepts(AC)","Unified Concept Editing(UCE)", "(Safe Latent Diffusion)SLD"],
34
+ label="Unlearned DMs")
35
+ with gr.Column():
36
+ # gr.Markdown("Please upload your model id.")
37
+ drop_text = gr.Dropdown(["Object-Church", "Object-Parachute", "Object-Garbage_Truck","Object-Tench",
38
+ "Style-Van_Gogh","Concept-Nudity", "None"],
39
+ label="AdvUnlearn Text Encoder")
40
 
41
  with gr.Row() as attack:
42
  text_input = gr.Textbox(label="Prompt")
43
+ with gr.Row():
44
+ with gr.Column(min_width=260):
45
+
46
+ img1 = gr.Image("images/cheetah.jpg",label="Image Generated without AdvUnlearn",width=260,show_share_button=False,show_download_button=False)
47
+ with gr.Column():
48
+ start_button = gr.Button("AdvUnlearn",size='lg')
49
+ with gr.Column(min_width=260):
50
+ text_ouput = gr.Textbox(label="Prompt Genetated by UnlearnDiffAtk")
51
+ img2 = gr.Image("images/cheetah.jpg",label="Image Generated without AdvUnlearn",width=260,show_share_button=False,show_download_button=False)
52
 
53
  demo.launch()