Spaces:
Sleeping
Sleeping
[Update]Adjust indent
Browse files
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 |
-
|
32 |
-
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
|
40 |
|
41 |
with gr.Row() as attack:
|
42 |
text_input = gr.Textbox(label="Prompt")
|
43 |
-
with gr.
|
44 |
-
|
45 |
-
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
-
|
|
|
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()
|