torinriley commited on
Commit
32478f9
1 Parent(s): f132f1f

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -61,7 +61,7 @@ class EndpointHandler:
61
  results = []
62
  for box, label, score in zip(boxes, labels, scores):
63
  if score >= CONFIDENCE_THRESHOLD:
64
- x1, y1, x2, y2 = map(int, box)
65
  label_text = self.label_map.get(label, "unknown")
66
  results.append({
67
  "label": label_text,
 
61
  results = []
62
  for box, label, score in zip(boxes, labels, scores):
63
  if score >= CONFIDENCE_THRESHOLD:
64
+ x1, y1, x2, y2 = map(int, box) # Ensure integers for box coordinates
65
  label_text = self.label_map.get(label, "unknown")
66
  results.append({
67
  "label": label_text,