should work

handler.py

@@ -14,16 +14,15 @@ class EndpointHandler():
         self.resnet = InceptionResnetV1(pretrained='vggface2', device=self.device).eval()
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
-        imageData = data.pop("image", data)
-        return imageData
-        # 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]
-        # if face_batch:
-        #     aligned = torch.stack(face_batch)
-        #     if self.device.type == "cuda":
-        #         aligned = aligned.to(self.device)
+        imageData = data.get("inputs").get("image")
+        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]
+        if face_batch:
+            aligned = torch.stack(face_batch)
+            if self.device.type == "cuda":
+                aligned = aligned.to(self.device)
 
-        #     embeddings = self.resnet(aligned).detach().cpu()
-        #     return embeddings.tolist()
-        # else: return None
+            embeddings = self.resnet(aligned).detach().cpu()
+            return embeddings.tolist()
+        else: return None

test.py

@@ -7,6 +7,8 @@ inputs = {
 }
 
 
-result = my_handler(inputs)
+result = my_handler({
+    "inputs": inputs
+})
 
 print(result)