image in the right place

handler.py

@@ -14,7 +14,7 @@ class EndpointHandler():
         self.resnet = InceptionResnetV1(pretrained='vggface2', device=self.device).eval()
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
-        imageData = data['image']
+        imageData = data.pop("image", data)
         image = Image.open(BytesIO(base64.b64decode(imageData)))
         face_batch = self.mtcnn([image])
         face_batch = [i for i in face_batch if i is not None]