Update app.py

app.py

@@ -343,6 +343,8 @@ def predict(input):
     print(f'liuyz_0_', input)  
     image_np = np.array(input["image"])
     print(f'image_np = {image_np.shape}')
+    mask_np = np.array(input["mask"])
+    print(f'mask_np = {mask_np.shape}')    
     '''
     image = dict["image"] # .convert("RGB") #.resize((512, 512))
     # target_size = (init_image.shape[0], init_image.shape[1])
@@ -355,7 +357,7 @@ def predict(input):
     
     # output = mask #output.images[0]
     # output = pipe(prompt = prompt, image=init_image, mask_image=mask,guidance_scale=7.5)
-    output = None
+    output = input["mask"]
     return output #, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
 
 print(f'liuyz_500_here_')