added custom handler

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+# GitHub syntax highlighting
+pixi.lock linguist-language=YAML linguist-generated=true

.gitignore

@@ -0,0 +1,3 @@
+# pixi environments
+.pixi
+*.egg-info

handler.py

@@ -0,0 +1,12 @@
+from typing import Dict, List, Any
+from retinaface import RetinaFace
+import numpy as np
+from PIL import Image
+
+class EndpointHandler():
+    def __init__(self, path=""):
+        return
+        
+    def __call__(self, data: np.ndarray) -> List[Dict[str, Any]]:
+       res = RetinaFace.detect_faces(data)
+       return [res]

pixi.toml

@@ -0,0 +1,16 @@
+[project]
+name = "retinaface"
+version = "0.1.0"
+description = "Add a short description here"
+authors = ["EdvardsZ <edvards1999@gmail.com>"]
+channels = ["conda-forge"]
+platforms = ["linux-64"]
+
+[tasks]
+
+[dependencies]
+python = "3.11.*"
+transformers = ">=4.41.2,<4.42"
+retina-face = ">=0.0.17,<0.1"
+tf-keras = ">=2.16.0,<2.17"
+matplotlib = ">=3.8.4,<3.9"

test.py

@@ -0,0 +1,23 @@
+from handler import EndpointHandler
+from PIL import Image
+import numpy as np
+import matplotlib.pyplot as plt
+
+# init handler
+my_handler = EndpointHandler(path=".")
+
+# load image
+img = Image.open("angelina.jpg")
+
+img_np = np.array(img)
+
+# call handler
+result = my_handler(img_np)
+
+print(result)
+
+
+
+
+
+