Update app.py

app.py

@@ -127,26 +127,25 @@ def model_process(image, mask):
     # RGB
     # origin_image_bytes = input["image"].read()
     
-    print(f'liuyz_2_here_', type(image), image)
-    
-    image_pil = Image.fromarray(np.array(image))
-    mask_pil = Image.fromarray(np.array(mask)).convert("L")
+    print(f'liuyz_2_here_', type(image), image.shape)
+        
+    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') 
+    mask_pil.save(f'./mask_pil.png')         
         
-        
-    image, alpha_channel = load_img(image)
+    #image, alpha_channel = load_img(image)
     # Origin image shape: (512, 512, 3)
     
-    # alpha_channel = None
+    alpha_channel = None
     original_shape = image.shape
     interpolation = cv2.INTER_CUBIC    
     
     # form = request.form
     print(f'liuyz_3_here_', original_shape, alpha_channel)
 
-    size_limit = image_pil.shape[1] # : Union[int, str] = form.get("sizeLimit", "1080")
+    size_limit = "Original" # image.shape[1] # : Union[int, str] = form.get("sizeLimit", "1080")
     if size_limit == "Original":
         size_limit = max(image.shape)
     else:
@@ -188,8 +187,8 @@ def model_process(image, mask):
     # logger.info(f"Resized image shape: {image.shape}")
     print(f"Resized image shape: {image.shape} / {image[250][250]}")
     
-    mask, _ = load_img(mask, gray=True)
-    # mask = np.array(mask_pil)
+    # mask, _ = load_img(mask, gray=True)
+    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]}")