Update app.py

app.py

@@ -94,9 +94,8 @@ def get_predictions(y_prediction_encoded):
 
 def predict_on_train(image, mask):
 
-    # Get the resized image
+    # Get the resized image and mask
     sample_image_resized = resize_image(image)
-    # Get the resized mask
     mask_resized = resize_image(mask)
 
     # Create a figure