Update app.py

app.py

@@ -128,11 +128,13 @@ def model_process(image, mask):
     # origin_image_bytes = input["image"].read()
     
     print(f'liuyz_2_here_')
+    
     image_pil = Image.fromarray(image)
     mask_pil = Image.fromarray(mask).convert("L")
-    print(f'image_pil_ = {image_pil.shape}')
-    print(f'mask_pil_ = {mask_pil.shape}') 
-    mask_pil.save(f'./mask_pil.png')     
+    print(f'image_pil_ = {type(image_pil)}')
+    print(f'mask_pil_ = {type(mask_pil)}') 
+    mask_pil.save(f'./mask_pil.png') 
+        
         
     image, alpha_channel = load_img(image)
     # Origin image shape: (512, 512, 3)