border colour

app.py

@@ -3,7 +3,7 @@ os.system('pip install gradio==4.29.0')
 
 import random
 from dataclasses import dataclass
-from typing import Any, List, Dict, Optional, Union, Tuple
+from typing: Any, List, Dict, Optional, Union, Tuple
 import cv2
 import torch
 import requests
@@ -55,9 +55,9 @@ 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()
 
         if include_bboxes:
+            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),
                         cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, 2)
@@ -65,7 +65,7 @@ def annotate(image: Union[Image.Image, np.ndarray], detection_results: List[Dete
         if mask is not None:
             mask_uint8 = (mask * 255).astype(np.uint8)
             contours, _ = cv2.findContours(mask_uint8, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-            cv2.drawContours(image_cv2, contours, -1, color, 2)
+            cv2.drawContours(image_cv2, contours, -1, (0, 255, 0), 2)
 
     return cv2.cvtColor(image_cv2, cv2.COLOR_BGR2RGB)