Update app.py

app.py

@@ -55,6 +55,8 @@ if uploaded_image:
         # Create an empty label map for ranking based on area
         label_map = np.zeros_like(colorized_img)
 
+        _, binary_map = cv2.threshold(intensity_map, 255, 0, cv2.THRESH_BINARY)
+
         # Overlay the labels on the blended image
         font = cv2.FONT_HERSHEY_SIMPLEX
         contours, _ = cv2.findContours(binary_map, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)