Update app.py

app.py

@@ -212,8 +212,8 @@ def read_content(file_path: str) -> str:
     return content
 
 def predict(dict):
-    '''
-    print(f'liuyz_0_', dict)    
+    print(f'liuyz_0_', 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)
@@ -223,8 +223,8 @@ def predict(dict):
     '''
     image = Image.fromarray(dict["image"])
     mask= Image.fromarray(dict["mask"])
-    output = model_process(image, mask)
-    output = output.images[0]
+    #output = model_process(image, mask)
+    output = mask #output.images[0]
     # output = pipe(prompt = prompt, image=init_image, mask_image=mask,guidance_scale=7.5)
     
     return output #, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
@@ -327,18 +327,20 @@ with image_blocks as demo:
                             rounded=(True, True, True, True),
                             full_width=True,
                         )                
-                '''
+                
                 with gr.Column():
-                    image_out = gr.Image(label="Output", elem_id="output-img").style(height=400)
+                    image_out = gr.Image(label="Output").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])
+            btn.click(fn=predict, inputs=[image], outputs=[image_out]) #, community_icon, loading_icon, share_button])
             #share_button.click(None, [], [], _js=share_js)
             
             gr.HTML(