From 8589f935884dc3863e2fefac34e1880406a251cf Mon Sep 17 00:00:00 2001
From: Yazhou Cao <yazhou.cao@landing.ai>
Date: Thu, 4 Apr 2024 15:55:15 -0700
Subject: [PATCH] Include scores in the G-SAM model response

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

diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
index 164ca6ea..51ae01ca 100644
--- a/vision_agent/tools/tools.py
+++ b/vision_agent/tools/tools.py
@@ -301,6 +301,7 @@ def __call__(self, prompt: str, image: Union[str, ImageType]) -> Dict:
                 for mask in data["masks"]
             ]
         ret_pred["labels"] = data["labels"]
+        ret_pred["scores"] = data["scores"]
         return ret_pred