fixed issue with space (#205)

landing-ai · Aug 21, 2024 · 62d4e8e · 62d4e8e
1 parent c9aaa68
commit 62d4e8e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",