yizhangliu commited on
Commit
4b39dfc
1 Parent(s): 46bf436

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -186,7 +186,7 @@ def model_process(image, mask):
186
  # logger.info(f"Resized image shape: {image.shape}")
187
  print(f"Resized image shape: {image.shape} / {image[250][250]}")
188
 
189
- mask, _ = load_img(mask), gray=True)
190
  # mask = np.array(mask_pil)
191
  mask = resize_max_size(mask, size_limit=size_limit, interpolation=interpolation)
192
  print(f"mask image shape: {mask.shape} / {type(mask)} / {mask[250][250]}")
 
186
  # logger.info(f"Resized image shape: {image.shape}")
187
  print(f"Resized image shape: {image.shape} / {image[250][250]}")
188
 
189
+ mask, _ = load_img(mask, gray=True)
190
  # mask = np.array(mask_pil)
191
  mask = resize_max_size(mask, size_limit=size_limit, interpolation=interpolation)
192
  print(f"mask image shape: {mask.shape} / {type(mask)} / {mask[250][250]}")