Update app.py

app.py

@@ -151,12 +151,11 @@ 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)
+        mask = cv2.imread(self.masks_fps, 0)
 
         mask[np.where(mask < 8)] = 0
         mask[np.where(mask > 8)] = 255
-#        extract certain classes from mask (e.g. polyp)
+        # extract certain classes from mask (e.g. polyp)
         masks = [(mask == v) for v in self.class_values]
         mask = np.stack(masks, axis=-1).astype('float')