Update app.py

app.py

@@ -85,7 +85,7 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=30, s
     print("Strength:", strength)
     print("Scheduler:", scheduler)
 
-    return inpainted_image, gr.update(visible=True)
+    return inpainted_image, combined_image, gr.update(visible=True)
 
 
 css = '''
@@ -150,14 +150,15 @@ with image_blocks as demo:
                         
                 with gr.Column():
                     image_out = gr.Image(label="Output", elem_id="output-img", height=512, width=512)
+                    image_out1 = gr.Image(label="Output", elem_id="output-img", height=512, width=512)
                     with gr.Group(elem_id="share-btn-container", visible=False) as share_btn_container:
                         community_icon = gr.HTML(community_icon_html)
                         loading_icon = gr.HTML(loading_icon_html)
                         share_button = gr.Button("Share to community", elem_id="share-btn",visible=True)
             
 
-    btn.click(fn=predict, inputs=[image, prompt, negative_prompt, guidance_scale, steps, strength, model, scheduler], outputs=[image_out,share_btn_container], api_name='run')
-    prompt.submit(fn=predict, inputs=[image, prompt, negative_prompt, guidance_scale, steps, strength, model, scheduler], outputs=[image_out,share_btn_container])
+    btn.click(fn=predict, inputs=[image, prompt, negative_prompt, guidance_scale, steps, strength, model, scheduler], outputs=[image_out,image_out1,share_btn_container], api_name='run')
+    prompt.submit(fn=predict, inputs=[image, prompt, negative_prompt, guidance_scale, steps, strength, model, scheduler], outputs=[image_out,image_out1,share_btn_container])
     share_button.click(None, [], [], _js=share_js)