Update app.py

app.py

@@ -480,7 +480,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)