Update app.py

app.py

@@ -54,19 +54,12 @@ def annotate(image: Union[Image.Image, np.ndarray], detection_results: List[Dete
     yellow_background = np.full(image_cv2.shape, (0, 255, 255), dtype=np.uint8)
     
     for detection in detection_results:
-        box = detection.box
         mask = detection.mask
         
         if mask is not None:
             mask_uint8 = (mask * 255).astype(np.uint8)
-            mask_inv = cv2.bitwise_not(mask_uint8)
-            
-            # Extract insect using mask
             insect_region = cv2.bitwise_and(image_cv2, image_cv2, mask=mask_uint8)
-            yellow_background_masked = cv2.bitwise_and(yellow_background, yellow_background, mask=mask_inv)
-            insect_combined = cv2.add(insect_region, yellow_background_masked)
-            
-            yellow_background = cv2.add(yellow_background, insect_combined)
+            yellow_background = np.where(mask[..., None], insect_region, yellow_background)
 
     return cv2.cvtColor(yellow_background, cv2.COLOR_BGR2RGB)
 
@@ -218,4 +211,4 @@ gr.Interface(
     outputs=[gr.Image(type="numpy"), gr.Textbox()],
     title="InsectSAM 🐞",
     examples=examples
-).launch()
+).launch()