From 81f3cf39d05d6f14e3d66d6e57431d201d8f4032 Mon Sep 17 00:00:00 2001
From: Dillon Laird <dillonalaird@gmail.com>
Date: Thu, 18 Apr 2024 11:32:02 -0700
Subject: [PATCH] fix color issue

---
 vision_agent/image_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vision_agent/image_utils.py b/vision_agent/image_utils.py
index d0164e11..3a34fe93 100644
--- a/vision_agent/image_utils.py
+++ b/vision_agent/image_utils.py
@@ -103,7 +103,7 @@ def overlay_bboxes(
     elif isinstance(image, np.ndarray):
         image = Image.fromarray(image)
 
-    color = {label: COLORS[i % len(COLORS)] for i, label in enumerate(bboxes["labels"])}
+    color = {label: COLORS[i % len(COLORS)] for i, label in enumerate(set(bboxes["labels"]))}
 
     width, height = image.size
     fontsize = max(12, int(min(width, height) / 40))