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)
 
+        intensity_map = cv2.cvtColor(colorized_img, cv2.COLOR_BGR2GRAY)
+        
         _, binary_map = cv2.threshold(intensity_map, 255, 0, cv2.THRESH_BINARY)
 
         # Overlay the labels on the blended image