Update app.py

app.py

@@ -355,7 +355,7 @@ with gr.Blocks(css=css) as demo:
         inputs=[input_image, width_slider, height_slider, overlap_percentage, num_inference_steps,
                 resize_option, custom_resize_percentage, prompt_input, alignment_dropdown,
                 overlap_left, overlap_right, overlap_top, overlap_bottom],
-        outputs=result,
+        outputs=[result, preview_image],
     ).then(
         fn=lambda x, history: update_history(x[1], history),
         inputs=[result, history_gallery],
@@ -374,7 +374,7 @@ with gr.Blocks(css=css) as demo:
         fn=inpaint,
         inputs=[input_image, width_slider, height_slider, overlap_percentage, num_inference_steps, resize_option, custom_resize_percentage, prompt_input, alignment_dropdown,
                 overlap_left, overlap_right, overlap_top, overlap_bottom],
-        outputs=result,
+        outputs=[result, preview_image],
     ).then(
         fn=lambda x, history: update_history(x[1], history),
         inputs=[result, history_gallery],