Spaces:
Build error
Build error
Update app.py
Browse files
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)
|