carlgira commited on
Commit
745568f
·
1 Parent(s): cf67a3b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -78,15 +78,22 @@ with gr.Blocks(title='Dreambooth Image Editing and Stable Diffusion Inpainting')
78
  with gr.Column():
79
  img_ctr = gr.Image(tool='select')
80
  with gr.Column():
81
- output = gr.Image()
82
  with gr.Row():
83
  greet_btn = gr.Button("Selection")
84
  with gr.Row():
85
  sd_prompt = gr.Textbox(lines=2, label="Stable diffusion prompt")
86
  with gr.Row():
87
- final_image = gr.Image()
88
  with gr.Row():
89
  stab_btn = gr.Button("Inpaint")
 
 
 
 
 
 
 
90
 
91
  greet_btn.click(fn=preview, inputs=[img_ctr, state], outputs=[output, state])
92
  stab_btn.click(fn=sd_inpaint, inputs=[output, sd_prompt, state], outputs=final_image)
 
78
  with gr.Column():
79
  img_ctr = gr.Image(tool='select')
80
  with gr.Column():
81
+ output = gr.Image(label="Selection with mask (512x512)")
82
  with gr.Row():
83
  greet_btn = gr.Button("Selection")
84
  with gr.Row():
85
  sd_prompt = gr.Textbox(lines=2, label="Stable diffusion prompt")
86
  with gr.Row():
87
+ final_image = gr.Image(label='Generated Image (512x512)')
88
  with gr.Row():
89
  stab_btn = gr.Button("Inpaint")
90
+ with gr.Row():
91
+ gr.Examples([
92
+ ['one.png', 'in the office'],
93
+ ], inputs=[img_ctr, sd_prompt])
94
+ gr.Examples([
95
+ ['two.png', 'in the office'],
96
+ ], inputs=[img_ctr, sd_prompt])
97
 
98
  greet_btn.click(fn=preview, inputs=[img_ctr, state], outputs=[output, state])
99
  stab_btn.click(fn=sd_inpaint, inputs=[output, sd_prompt, state], outputs=final_image)