Update app.py

app.py

@@ -151,7 +151,9 @@ class Dataset(BaseDataset):
         # read data
         image = cv2.imread(self.images_fps)
         image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
-        mask = cv2.imread(self.masks_fps, 0)
+#        mask = cv2.imread(self.masks_fps, 0)
+        mask = cv2.imread(self.masks_fps)
+
         mask[np.where(mask < 8)] = 0
         mask[np.where(mask > 8)] = 255
 #        extract certain classes from mask (e.g. polyp)
@@ -221,7 +223,6 @@ def model_infer(img_name):
         image, mask = input
 
         image_vis = image.transpose(1, 2, 0)
-        #gt_mask = mask[0]
         gt_mask = mask[0].astype('uint8')
         pr_mask = (output[0].numpy() > threshold).astype('uint8')[0]
         i = i + 1