Update app.py

app.py

@@ -90,7 +90,7 @@ def inpaint(ip, image_editor, prompt, neg_prompt, width, height, ip_scale, stren
         num_inference_steps=steps,
     ).images
     
-    return images[0]
+    return images[0], image, mask 
 
 
 with gr.Blocks() as demo:
@@ -116,6 +116,8 @@ with gr.Blocks() as demo:
             inpaint_button = gr.Button("Generate")
             
         output_image = gr.Image(label='Result')
+        output1_image = gr.Image(label='Result')
+        output2_image = gr.Image(label='Result')
         
     with gr.Accordion("Advanced Settings", open=False):
         neg_prompt = gr.Textbox(label='Negative Prompt', value='ugly, deformed, nsfw')
@@ -128,6 +130,6 @@ with gr.Blocks() as demo:
 
     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)
     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)
-    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)
+    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])
 
 demo.launch()