Update app.py

app.py

@@ -94,14 +94,19 @@ def get_predictions(y_prediction_encoded):
 
 def predict_on_train(image, mask):
 
-    # Steps to get prediction of the satellite image
+    # Get the resized image
+    sample_image_resized = resize_image(image)
+    # Get the resized mask
     mask_resized = resize_image(mask)
 
     # Create a figure
     fig, ax = plt.subplots()
 
     # Display the image
-    ax.imshow(mask_resized)
+    ax.imshow(sample_image_resized)
+    
+    # Display the image
+    cax = ax.imshow(mask_resized, alpha=0.5)
 
     # Convert the figure to a PIL Image
     image_buffer = io.BytesIO()
@@ -115,7 +120,6 @@ def predict_on_train(image, mask):
     # ----------------------------------------------
     
     # Steps to get prediction of the satellite image
-    sample_image_resized = resize_image(image)
     y_pred = ensemble_predict(sample_image_resized)
     y_pred = get_predictions(y_pred).squeeze()