From becc7f3f1c95aee507c3e17bc66fd9146d26faa0 Mon Sep 17 00:00:00 2001 From: Dayanne Fernandes Date: Mon, 2 Sep 2024 20:47:08 -0300 Subject: [PATCH] add model --- vision_agent/tools/tools.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py index afd4f6bb..960f51f9 100644 --- a/vision_agent/tools/tools.py +++ b/vision_agent/tools/tools.py @@ -540,7 +540,7 @@ def countgd_counting( """ buffer_bytes = numpy_to_bytes(image) files = [("image", buffer_bytes)] - payload = {"prompts": [prompt]} + payload = {"prompts": [prompt], "model": "countgd"} metadata = {"function_name": "countgd_counting"} resp_data: List[Dict[str, Any]] = send_task_inference_request( payload, "text-to-object-detection", files=files, metadata=metadata @@ -591,7 +591,7 @@ def countgd_example_based_counting( visual_prompts = [ denormalize_bbox(bbox, image.shape[:2]) for bbox in visual_prompts ] - payload = {"visual_prompts": json.loads(visual_prompts)} + payload = {"visual_prompts": json.loads(visual_prompts), "model": "countgd"} metadata = {"function_name": "countgd_example_based_counting"} resp_data: List[Dict[str, Any]] = send_task_inference_request( payload, "visual-prompts-to-object-detection", files=files, metadata=metadata