Kaixuanliu commited on
Commit
566b9eb
·
verified ·
1 Parent(s): c2be33a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -14
app.py CHANGED
@@ -15,7 +15,7 @@ from about import (
15
  TITLE,
16
  )
17
 
18
- myip = "0.0.0.0"
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
- # 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
- 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):