Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ from about import (
|
|
15 |
TITLE,
|
16 |
)
|
17 |
|
18 |
-
myip = "
|
19 |
myport=80
|
20 |
is_spaces = True if "SPACE_ID" in os.environ else False
|
21 |
is_shared_ui = False
|
@@ -25,24 +25,15 @@ with gr.Blocks() as demo:
|
|
25 |
gr.HTML(TITLE)
|
26 |
gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
|
27 |
with gr.Row() as advlearn:
|
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 |
-
|
33 |
-
|
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 |
-
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):
|
|
|
15 |
TITLE,
|
16 |
)
|
17 |
|
18 |
+
myip = "34.219.98.113"
|
19 |
myport=80
|
20 |
is_spaces = True if "SPACE_ID" in os.environ else False
|
21 |
is_shared_ui = False
|
|
|
25 |
gr.HTML(TITLE)
|
26 |
gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
|
27 |
with gr.Row() as advlearn:
|
|
|
|
|
|
|
28 |
with gr.Column():
|
29 |
+
# gr.Markdown("Please upload your model id.")
|
30 |
+
drop_text = gr.Dropdown(["Object-Church", "Object-Parachute", "Object-Garbage_Truck","Object-Tench",
|
|
|
|
|
|
|
|
|
31 |
"Style-Van_Gogh","Concept-Nudity", "None"],
|
32 |
label="AdvUnlearn Text Encoder")
|
33 |
+
with gr.Column():
|
34 |
+
text_input = gr.Textbox(label="Prompt")
|
35 |
|
|
|
|
|
36 |
with gr.Row():
|
|
|
|
|
37 |
with gr.Column():
|
38 |
start_button = gr.Button("AdvUnlearn",size='lg')
|
39 |
with gr.Column(min_width=260):
|