antoyo123 commited on
Commit
3fc5ae5
·
1 Parent(s): 5810a68

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ def process_image(image, prompt):
38
  image = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
39
  imageio.imwrite("./data/data.png", image)
40
  # Convert mask to grayscale format
41
- mask_image = mask_image.resize(image.size)
42
  mask_image = cv2.cvtColor(np.array(mask_image), cv2.COLOR_RGB2GRAY)
43
 
44
 
 
38
  image = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
39
  imageio.imwrite("./data/data.png", image)
40
  # Convert mask to grayscale format
41
+ mask_image = mask_image.resize((image.shape[1], image.shape[0]))
42
  mask_image = cv2.cvtColor(np.array(mask_image), cv2.COLOR_RGB2GRAY)
43
 
44