torinriley
commited on
Commit
•
32478f9
1
Parent(s):
f132f1f
Update handler.py
Browse files- 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,
|