English
detection
open-world
open-set
Inference Endpoints
kelvinou01 commited on
Commit
296053a
1 Parent(s): 13ea7e7
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -51,7 +51,7 @@ class EndpointHandler():
51
  annotated_frame = annotate(image_source=image_source, boxes=boxes, logits=logits, phrases=phrases)
52
  _, annotated_image = cv2.imencode(".jpg", annotated_frame)
53
  annotated_image_b64 = base64.b64encode(annotated_image).decode("utf-8")
54
- num_found = boxes.size[0]
55
 
56
  return [{
57
  "image": annotated_image_b64,
 
51
  annotated_frame = annotate(image_source=image_source, boxes=boxes, logits=logits, phrases=phrases)
52
  _, annotated_image = cv2.imencode(".jpg", annotated_frame)
53
  annotated_image_b64 = base64.b64encode(annotated_image).decode("utf-8")
54
+ num_found = boxes.size(0)
55
 
56
  return [{
57
  "image": annotated_image_b64,