Update handler.py

handler.py

@@ -15,15 +15,14 @@ class EndpointHandler():
             class_id = results[0].names[box.cls[0].item()]
             cords = box.xywh[0].tolist()
             cords = [round(x) for x in cords]
-            conf = round(box.conf[0].item(), 2)
+            conf = box.conf[0].item()
             prediction = {
-            "label": class_id,
-            "score": conf,
-            "box":{
-                "x": cords[0],
-                "y": cords[1],
-                "width": cords[2],
-                "height": cords[3]}
+            "x": round(cords[0]),
+            "y": round(cords[1]),
+            "width": round(cords[2]),
+            "height": round(cords[3]),
+            "confidence": conf,
+            "class": class_id
             }
             predictions.append(prediction)
         predictions_array = {"predictions": predictions}