revert to cv2.cvtColor fix

app.py

@@ -46,7 +46,7 @@ class DetectionResult:
             )
         )
 
-def annotate(image: Union[Image.Image, np.ndarray], detection_results: List[DetectionResult], color=(0, 255, 255)) -> np.ndarray:
+def annotate(image: Union[Image.Image, np.ndarray], detection_results: List[DetectionResult]) -> np.ndarray:
     image_cv2 = np.array(image) if isinstance(image, Image.Image) else image
     image_cv2 = cv2.cvtColor(image_cv2, cv2.COLOR_RGB2BGR)
 
@@ -55,6 +55,7 @@ def annotate(image: Union[Image.Image, np.ndarray], detection_results: List[Dete
         score = detection.score
         box = detection.box
         mask = detection.mask
+        color = np.random.randint(0, 256, size=3).tolist()
 
         cv2.rectangle(image_cv2, (box.xmin, box.ymin), (box.xmax, box.ymax), color, 2)
         cv2.putText(image_cv2, f'{label}: {score:.2f}', (box.xmin, box.ymin - 10),
@@ -154,9 +155,9 @@ def create_yellow_background_with_insects(image: np.ndarray, detections: List[De
     for detection in detections:
         if detection.mask is not None:
             extract_and_paste_insect(image, detection, yellow_background)
-    # Draw bounding boxes and labels on yellow background
-    annotated_background = annotate(yellow_background, detections, color=(0, 255, 255))
-    return annotated_background
+    # Convert back to RGB to match Gradio's expected input format
+    yellow_background = cv2.cvtColor(yellow_background, cv2.COLOR_BGR2RGB)
+    return yellow_background
 
 def run_length_encoding(mask):
     pixels = mask.flatten()
@@ -215,4 +216,4 @@ gr.Interface(
     outputs=[gr.Image(type="numpy"), gr.Textbox()],
     title="InsectSAM 🐞",
     examples=examples
-).launch()
+).launch()