yizhangliu commited on
Commit
22e2ed0
·
1 Parent(s): 4a29657

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -117,6 +117,7 @@ def model_process(image, mask):
117
 
118
  # image, alpha_channel = load_img(origin_image_bytes)
119
  # Origin image shape: (512, 512, 3)
 
120
  original_shape = image.shape
121
  interpolation = cv2.INTER_CUBIC
122
 
 
117
 
118
  # image, alpha_channel = load_img(origin_image_bytes)
119
  # Origin image shape: (512, 512, 3)
120
+ alpha_channel = None
121
  original_shape = image.shape
122
  interpolation = cv2.INTER_CUBIC
123