Drazcat-AI commited on
Commit
39c617a
·
verified ·
1 Parent(s): f0775bb

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +3 -3
handler.py CHANGED
@@ -17,12 +17,12 @@ class EndpointHandler():
17
  cords = [round(x) for x in cords]
18
  conf = round(box.conf[0].item(), 2)
19
  prediction = {
20
- "class": class_id,
21
- "confidence": conf,
22
  "x": cords[0],
23
  "y": cords[1],
24
  "width": cords[2],
25
- "height": cords[3]
 
 
26
  }
27
  predictions.append(prediction)
28
  predictions_array = {"predictions": predictions}
 
17
  cords = [round(x) for x in cords]
18
  conf = round(box.conf[0].item(), 2)
19
  prediction = {
 
 
20
  "x": cords[0],
21
  "y": cords[1],
22
  "width": cords[2],
23
+ "height": cords[3],
24
+ "confidence": conf,
25
+ "class": class_id
26
  }
27
  predictions.append(prediction)
28
  predictions_array = {"predictions": predictions}