Update app.py

app.py

@@ -216,8 +216,8 @@ def predict(dict):
     init_image = dict["image"] # .convert("RGB") #.resize((512, 512))
     target_size = (init_image.shape[0], init_image.shape[1])
     print(f'liuyz_1_', init_image, target_size)
-    print(f'liuyz_2_', init_image.convert("RGB"))
-    print(f'liuyz_3_', init_image.convert("RGB").resize((512, 512)))
+    print(f'liuyz_2_', init_image.convert("RGB").shape)
+    print(f'liuyz_3_', init_image.convert("RGB").resize((512, 512)).shape)
     mask = dict["mask"] # .convert("RGB") #.resize((512, 512))
     # output = model_process(init_image, mask)
     output = None # output.images[0]
@@ -317,23 +317,25 @@ with image_blocks as demo:
                 with gr.Column():
                     image = gr.Image(source='upload', tool='sketch', elem_id="image_upload", label="Upload") #.style(height=400)
                     with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
-                        prompt = gr.Textbox(placeholder = 'Your prompt (what you want in place of what is erased)', show_label=False, elem_id="input-text")
+                        # prompt = gr.Textbox(placeholder = 'Your prompt (what you want in place of what is erased)', show_label=False, elem_id="input-text")
                         btn = gr.Button("Done!").style(
                             margin=True,
                             rounded=(True, True, True, True),
                             full_width=True,
                         )                
+                '''
                 with gr.Column():
                     image_out = gr.Image(label="Output", elem_id="output-img").style(height=400)
                     with gr.Group(elem_id="share-btn-container"):
                         community_icon = gr.HTML(community_icon_html, visible=False)
                         loading_icon = gr.HTML(loading_icon_html, visible=False)
                         share_button = gr.Button("Share to community", elem_id="share-btn", visible=False)
+                '''
                 
 
-            btn.click(fn=predict, inputs=[image, prompt], 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)
+            # btn.click(fn=predict, inputs=[image, prompt], 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)
             
             gr.HTML(
                 """