load mode local root

antelopev2/1k3d68.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:df5c06b8a0c12e422b2ed8947b8869faa4105387f199c477af038aa01f9a45cc
+size 143607619

antelopev2/2d106det.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f001b856447c413801ef5c42091ed0cd516fcd21f2d6b79635b1e733a7109dbf
+size 5030888

antelopev2/genderage.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4fde69b1c810857b88c64a335084f1c3fe8f01246c9a191b48c7bb756d6652fb
+size 1322532

antelopev2/glintr100.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4ab1d6435d639628a6f3e5008dd4f929edf4c4124b1a7169e1048f9fef534cdf
+size 260665334

antelopev2/scrfd_10g_bnkps.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5838f7fe053675b1c7a08b633df49e7af5495cee0493c7dcf6697200b85b5b91
+size 16923827

handler.py

@@ -144,10 +144,15 @@ class EndpointHandler():
         # model_pack_name = 'antelopev2' 
         # self.app = FaceAnalysis(allowed_modules=['detection', 'recognition'],name=model_pack_name)
         # self.app.prepare(ctx_id=0, det_size=(640, 640)) 
+
+        self.app = FaceAnalysis(name='antelopev2', root='./', providers=['CPUExecutionProvider'])
+        self.app.prepare(ctx_id=0, det_size=(640, 640))
+
         
-        self.app = FaceAnalysis(name="buffalo_l", root="./", providers=["CPUExecutionProvider"])
+        # self.app = FaceAnalysis(name="buffalo_l", root="./", providers=["CPUExecutionProvider"])
         # con 512 funciona bien
-        self.app.prepare(ctx_id=0, det_size=(640, 640))
+        # self.app.prepare(ctx_id=0, det_size=(640, 640))
+
         self.generator = torch.Generator(device=device).manual_seed(42)
 
 

requirements.txt

@@ -5,6 +5,7 @@ transformers==4.37.1
 accelerate
 safetensors
 einops
+onnxruntime
 onnxruntime-gpu
 spaces==0.19.4
 omegaconf