fixed extra space

landing-ai · Aug 21, 2024 · 8a36308 · 8a36308
1 parent c8a36fc
commit 8a36308
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
@@ -283,7 +283,7 @@ def florence2_sam2_image(prompt: str, image: np.ndarray) -> List[Dict[str, Any]]
 
  files = [("image", buffer_bytes)]
  payload = {
- "prompts": prompt.split(","),
+ "prompts": [s.strip() for s in prompt.split(",")],
  "function_name": "florence2_sam2_image",
  }
  data: Dict[str, Any] = send_inference_request(