From 62d4e8e089a4f3317dbc7a5f8f878eeb5ce99314 Mon Sep 17 00:00:00 2001
From: Dillon Laird <dillonalaird@gmail.com>
Date: Tue, 20 Aug 2024 17:06:30 -0700
Subject: [PATCH] fixed issue with space (#205)

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

diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
index a8339df8..0254a455 100644
--- a/vision_agent/tools/tools.py
+++ b/vision_agent/tools/tools.py
@@ -157,7 +157,7 @@ def owl_v2(
     image_size = image.shape[:2]
     image_b64 = convert_to_b64(image)
     request_data = {
-        "prompts": prompt.split(","),
+        "prompts": [s.strip() for s in prompt.split(",")],
         "image": image_b64,
         "confidence": box_threshold,
         "function_name": "owl_v2",