Update app.py
Browse files
app.py
CHANGED
@@ -94,9 +94,8 @@ def get_predictions(y_prediction_encoded):
|
|
94 |
|
95 |
def predict_on_train(image, mask):
|
96 |
|
97 |
-
# Get the resized image
|
98 |
sample_image_resized = resize_image(image)
|
99 |
-
# Get the resized mask
|
100 |
mask_resized = resize_image(mask)
|
101 |
|
102 |
# Create a figure
|
|
|
94 |
|
95 |
def predict_on_train(image, mask):
|
96 |
|
97 |
+
# Get the resized image and mask
|
98 |
sample_image_resized = resize_image(image)
|
|
|
99 |
mask_resized = resize_image(mask)
|
100 |
|
101 |
# Create a figure
|