Update handler.py

handler.py

@@ -43,7 +43,8 @@ class EndpointHandler:
         """
         try:
             if "body" not in data:
-                return {"error": "No image data provided in request."}
+                # Return error in the expected output format
+                return [{"error": "No image data provided in request."}]
 
             image_bytes = data["body"] 
             image_tensor = self.preprocess_frame(image_bytes)
@@ -51,10 +52,12 @@ class EndpointHandler:
             with torch.no_grad():
                 predictions = self.model(image_tensor)
 
+            # Extract predictions
             boxes = predictions[0]["boxes"].cpu().tolist()
             labels = predictions[0]["labels"].cpu().tolist()
             scores = predictions[0]["scores"].cpu().tolist()
 
+            # Build the results array
             results = []
             for box, label, score in zip(boxes, labels, scores):
                 if score >= CONFIDENCE_THRESHOLD:
@@ -71,6 +74,8 @@ class EndpointHandler:
                         }
                     })
 
+            # Return results in the required schema
             return results 
         except Exception as e:
-            return {"error": str(e)}
+            # Return errors wrapped in a list to match the required schema
+            return [{"error": str(e)}]