Update app.py

app.py

@@ -167,7 +167,9 @@ def model_process(image, mask):
     print(f"Resized image shape: {image.shape} / {image[250][250]}")
     
     #mask, _ = load_img(input["mask"].read(), gray=True)
-    mask = np.array(Image.fromarray(mask).convert("L"))
+    mask_image = Image.fromarray(mask).convert("L")
+    mask_image.save(f'./mask_image.png')
+    mask = np.array(mask_image)
     mask = resize_max_size(mask, size_limit=size_limit, interpolation=interpolation)
     print(f"mask image shape: {mask.shape} / {type(mask)} / {mask[250][250]}")
 
@@ -180,8 +182,9 @@ def model_process(image, mask):
     print(f"process time: {(time.time() - start) * 1000}ms, {res_np_img.shape} / {res_np_img[250][250]}")
 
     torch.cuda.empty_cache()
-
-    return Image.fromarray(res_np_img)
+    image = Image.fromarray(res_np_img)
+    image.save(f'./result_image.png')
+    return image
     '''
     if alpha_channel is not None:
         if alpha_channel.shape[:2] != res_np_img.shape[:2]: