app.py

app.py

@@ -21,7 +21,7 @@ def predict(image_file, segmentation_png, bitmap_img):
 
   label_file = tf.image.decode_bmp(tf.io.read_file(image_list[2]), channels=0)
   # label_file = tf.image.rgb_to_grayscale(tf.image.decode_bmp(tf.io.read_file(image_list[2]), channels=3))
-  #print("after decode_bmp --> ", label_file.shape, type(label_file))
+  # print("after decode_bmp --> ", label_file.shape, type(label_file))
   label_file = tf.squeeze(label_file)
 
   image_list = [segmentation_file, image, label_file]