Update app.py

app.py

@@ -126,15 +126,14 @@ def model_process_pil(input):
     # input = request.files
     # RGB
     # origin_image_bytes = input["image"].read()
-    image = input['image']
-    mask = input['mask']
-    print(f'liuyz_2_here_', type(image), image.shape)
+    image_pil = input['image']
+    mask_pil = input['mask']
         
-    image_pil = Image.fromarray(image)
-    mask_pil = Image.fromarray(mask).convert("L")
-    print(f'image_pil_ = {type(image_pil)}')
-    print(f'mask_pil_ = {type(mask_pil)}') 
-    mask_pil.save(f'./mask_pil.png')         
+    image = np.array(image_pil)
+    mask = np.array(mask_pil.convert("L"))
+    # 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)
@@ -189,7 +188,7 @@ def model_process_pil(input):
     print(f"Resized image shape: {image.shape} / {image[250][250]} / {image.dtype}")
     
     # mask, _ = load_img(mask, gray=True)
-    mask = np.array(mask_pil)
+    #mask = np.array(mask_pil)
     mask = resize_max_size(mask, size_limit=size_limit, interpolation=interpolation)
     print(f"mask image shape: {mask.shape} / {type(mask)} / {mask[250][250]} / {mask.dtype}")