should work

handler.py

@@ -15,15 +15,14 @@ class EndpointHandler():
 
     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)
+        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