Update app.py

app.py

@@ -109,7 +109,7 @@ with image_blocks as demo:
                 '''
 
             # btn.click(fn=predict, inputs=[image, prompt], outputs=[image_out, community_icon, loading_icon, share_button])
-            btn.click(fn=predict, inputs=[image], outputs=[image_out]) #, community_icon, loading_icon, share_button])
+            btn.click(fn=predict, inputs=[image], outputs=[image]) #, community_icon, loading_icon, share_button])
             # share_button.click(None, [], [], _js=share_js)
 
 image_blocks.launch()