Update app.py

app.py

@@ -223,7 +223,7 @@ def model_process(input): #image, mask):
     # {'image': '/tmp/tmp8mn9xw93.png', 'mask': '/tmp/tmpn5ars4te.png'}
     # input = request.files
     # RGB
-    origin_image_bytes = open(input["image"],'rb').read()
+    origin_image_bytes = read_content(input["image"])
     print(f'origin_image_bytes = ', type(origin_image_bytes), len(origin_image_bytes))
 
     image, alpha_channel = load_img(origin_image_bytes)
@@ -274,9 +274,9 @@ def model_process(input): #image, mask):
     logger.info(f"Resized image shape: {image.shape} / {type(image)}")
     print(f"Resized image shape: {image.shape} / {image[250][250]}")
 
-    mask, _ = load_img(open(input["mask"],'rb').read(), gray=True)
+    mask, alpha_channel = load_img(read_content(input["mask"]), gray=True)
     mask = resize_max_size(mask, size_limit=size_limit, interpolation=interpolation)
-    print(f"mask image shape: {mask.shape} / {type(mask)} / {mask[250][250]}")
+    print(f"mask image shape: {mask.shape} / {type(mask)} / {mask[250][250]} / {alpha_channel}")
 
     start = time.time()
     res_np_img = model(image, mask, config)
@@ -331,20 +331,22 @@ transform = transforms.Compose([
 ])
 '''
 
-def read_content(file_path: str) -> str:
+def read_content(file_path: str):
     """read the content of target file
     """
-    with open(file_path, 'r', encoding='utf-8') as f:
+    with open(file_path, 'rb', encoding='utf-8') as f:
         content = f.read()
 
     return content
 
 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}')    
+    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])
@@ -353,11 +355,11 @@ def predict(input):
     print(f'liuyz_3_', image.convert("RGB").resize((512, 512)).shape)
     # mask = dict["mask"] # .convert("RGB") #.resize((512, 512))
     '''   
-    # output = model_process(input) # dict["image"], dict["mask"])
+    output = model_process(input) # dict["image"], dict["mask"])
     
     # output = mask #output.images[0]
     # output = pipe(prompt = prompt, image=init_image, mask_image=mask,guidance_scale=7.5)
-    output = input["mask"]
+    # output = input["mask"]
     return output #, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
 
 print(f'liuyz_500_here_')