Drazcat-AI commited on
Commit
2fd8bdc
·
verified ·
1 Parent(s): 1618d72

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -20,7 +20,7 @@ class EndpointHandler():
20
  class_id = results[0].names[box.cls[0].item()]
21
  cords = box.xywh[0].tolist()
22
  cords = [round(x) for x in cords]
23
- conf = round(box.conf[0].item(), 2)
24
  prediction = {
25
  "x": cords[0],
26
  "y": cords[1],
 
20
  class_id = results[0].names[box.cls[0].item()]
21
  cords = box.xywh[0].tolist()
22
  cords = [round(x) for x in cords]
23
+ conf = box.conf[0].item()
24
  prediction = {
25
  "x": cords[0],
26
  "y": cords[1],