tonyassi commited on
Commit
5a6578b
·
verified ·
1 Parent(s): 56b27fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -91,7 +91,7 @@ def inpaint(ip, image_editor, prompt, neg_prompt, width, height, ip_scale, stren
91
  num_inference_steps=steps,
92
  ).images
93
 
94
- return images[0], image, mask
95
 
96
 
97
  with gr.Blocks() as demo:
@@ -117,8 +117,6 @@ with gr.Blocks() as demo:
117
  inpaint_button = gr.Button("Generate")
118
 
119
  output_image = gr.Image(label='Result')
120
- output1_image = gr.Image(label='Result')
121
- output2_image = gr.Image(label='Result')
122
 
123
  with gr.Accordion("Advanced Settings", open=False):
124
  neg_prompt = gr.Textbox(label='Negative Prompt', value='ugly, deformed, nsfw')
@@ -131,6 +129,6 @@ with gr.Blocks() as demo:
131
 
132
  text_button.click(text_to_image, inputs=[text_ip, text_prompt, neg_prompt, width_slider, height_slider, ip_scale_slider, strength_slider, guidance_slider, steps_slider], outputs=output_image)
133
  image_button.click(image_to_image, inputs=[image_ip, image_image, image_prompt, neg_prompt, width_slider, height_slider, ip_scale_slider, strength_slider, guidance_slider, steps_slider], outputs=output_image)
134
- inpaint_button.click(inpaint, inputs=[inpaint_ip, inpaint_editor, inpaint_prompt, neg_prompt, width_slider, height_slider, ip_scale_slider, strength_slider, guidance_slider, steps_slider], outputs=[output_image,output1_image,output2_image])
135
 
136
  demo.launch()
 
91
  num_inference_steps=steps,
92
  ).images
93
 
94
+ return images[0]
95
 
96
 
97
  with gr.Blocks() as demo:
 
117
  inpaint_button = gr.Button("Generate")
118
 
119
  output_image = gr.Image(label='Result')
 
 
120
 
121
  with gr.Accordion("Advanced Settings", open=False):
122
  neg_prompt = gr.Textbox(label='Negative Prompt', value='ugly, deformed, nsfw')
 
129
 
130
  text_button.click(text_to_image, inputs=[text_ip, text_prompt, neg_prompt, width_slider, height_slider, ip_scale_slider, strength_slider, guidance_slider, steps_slider], outputs=output_image)
131
  image_button.click(image_to_image, inputs=[image_ip, image_image, image_prompt, neg_prompt, width_slider, height_slider, ip_scale_slider, strength_slider, guidance_slider, steps_slider], outputs=output_image)
132
+ inpaint_button.click(inpaint, inputs=[inpaint_ip, inpaint_editor, inpaint_prompt, neg_prompt, width_slider, height_slider, ip_scale_slider, strength_slider, guidance_slider, steps_slider], outputs=output_image)
133
 
134
  demo.launch()