Update app.py

app.py

@@ -227,8 +227,8 @@ def predict(input):
 
 css = '''
 .container {max-width: 1150px;margin: auto;padding-top: 1.5rem}
-image_upload{min-height:512px}
-image_upload [data-testid="image"], #image_upload [data-testid="image"] > div{min-height: 512px}
+#image_upload{min-height:512px}
+#image_upload [data-testid="image"], #image_upload [data-testid="image"] > div{min-height: 512px}
 #mask_radio .gr-form{background:transparent; border: none}
 #word_mask{margin-top: .75em !important}
 #word_mask textarea:disabled{opacity: 0.3}
@@ -322,8 +322,8 @@ with image_blocks as demo:
                             full_width=True,
                         )                
                 
-                # with gr.Column():
-                    image_out = gr.Image(label="Output", elem_id="image_output", visible=False) #.style(height=512)
+                with gr.Column():
+                    image_out = gr.Image(label="Output", elem_id="image_output", visible=True) #.style(height=512)
                     '''
                     with gr.Group(elem_id="share-btn-container"):
                         community_icon = gr.HTML(community_icon_html, visible=False)
@@ -334,7 +334,7 @@ with image_blocks as demo:
                 
 
             # btn.click(fn=predict, inputs=[image, prompt], outputs=[image_out, community_icon, loading_icon, share_button])
-            btn_in.click(fn=predict, inputs=[image], outputs=[image]) #, community_icon, loading_icon, share_button])
+            btn_in.click(fn=predict, inputs=[image], outputs=[image_out]) #, community_icon, loading_icon, share_button])
             #share_button.click(None, [], [], _js=share_js)