Update handler.py

handler.py

@@ -17,13 +17,12 @@ class EndpointHandler():
             cords = [round(x) for x in cords]
             conf = round(box.conf[0].item(), 2)
             prediction = {
-            "label": class_id,
-            "score": conf,
-            "box":{
-                "x": cords[0],
-                "y": cords[1],
-                "width": cords[2],
-                "height": cords[3]}
+            "class": class_id,
+            "confidence": conf,
+            "x": cords[0],
+            "y": cords[1],
+            "width": cords[2],
+            "height": cords[3]
             }
             predictions.append(prediction)
         predictions_array = {"predictions": predictions}