Add num_found

handler.py

@@ -13,9 +13,7 @@ HOME = os.getcwd()
 
 # /opt/conda/lib/python3.9/site-packages/groundingdino
 PACKAGE_HOME = os.path.dirname(groundingdino.__file__)
-
 CONFIG_PATH = os.path.join(PACKAGE_HOME, "config", "GroundingDINO_SwinT_OGC.py")
-WEIGHTS_PATH = os.path.join("model", "weights", "groundingdino_swint_ogc.pth")
 
 class EndpointHandler():
     def __init__(self, path):
@@ -53,8 +51,10 @@ class EndpointHandler():
             annotated_frame = annotate(image_source=image_source, boxes=boxes, logits=logits, phrases=phrases)
             _, annotated_image = cv2.imencode(".jpg", annotated_frame)
             annotated_image_b64 = base64.b64encode(annotated_image).decode("utf-8")
+            num_found = boxes.size[0]
 
             return [{
                 "image": annotated_image_b64,
                 "prompt": prompt,
+                "num_found": num_found,
             }]