From 841a4ae73602be46a131271aa17beaf29ba2996b Mon Sep 17 00:00:00 2001
From: Jon Elling <jonathanelling@gmail.com>
Date: Wed, 12 Feb 2025 18:04:04 -0500
Subject: [PATCH] return parsed data for grounded phrase

---
 nodes.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index f83351d..0d47cdc 100644
--- a/nodes.py
+++ b/nodes.py
@@ -454,7 +454,10 @@ class Florence2Run:
                 out_tensor = annotated_image_tensor[:3, :, :].unsqueeze(0).permute(0, 2, 3, 1).cpu().float()
                 out.append(out_tensor)
                
-                out_data.append(bboxes)
+                if task == 'caption_to_phrase_grounding':
+                    out_data.append(parsed_answer[task_prompt])
+                else:
+                    out_data.append(bboxes)
 
                 
                 pbar.update(1)