test

handler.py

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