antoyo123 commited on
Commit
79eb32b
·
1 Parent(s): 57de9a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def process_image(image, prompt):
34
  plt.imsave("mask.png", torch.sigmoid(preds))
35
  mask_image = Image.open("mask.png").convert("RGB")
36
 
37
- image = cv2.cvtColor(np.array(image))
38
  imageio.imwrite("./data/data.png", image)
39
  # Convert mask to grayscale format
40
  mask_image = mask_image.resize((image.shape[1], image.shape[0]))
 
34
  plt.imsave("mask.png", torch.sigmoid(preds))
35
  mask_image = Image.open("mask.png").convert("RGB")
36
 
37
+ image = np.array(image)
38
  imageio.imwrite("./data/data.png", image)
39
  # Convert mask to grayscale format
40
  mask_image = mask_image.resize((image.shape[1], image.shape[0]))