Update app.py

app.py

@@ -775,7 +775,7 @@ def process_images(input_image=None, input_tiff=None, gt_mask_file=None):
             ax_idx += 1
 
         # Ensure pred_mask is suitable for imshow (e.g., scale if needed, but should be 0/1)
-        axes[ax_idx].imshow(pred_mask, cmap='plasma') # Use plasma for Prediction
+        axes[ax_idx].imshow(pred_mask, cmap='viridis') # Use plasma for Prediction
         axes[ax_idx].set_title("Predicted Wetlands (128x128)")
         axes[ax_idx].axis('off')