test

handler.py

@@ -145,13 +145,12 @@ class EndpointHandler():
         # 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='antelopev2', root='./', providers=['CPUExecutionProvider'])
+        # self.app.prepare(ctx_id=0, det_size=(640, 640))
 
         
-        # 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 = FaceAnalysis(name="buffalo_l", root="./", providers=["CPUExecutionProvider"])
+        self.app.prepare(ctx_id=0, det_size=(640, 640))
 
         self.generator = torch.Generator(device=device).manual_seed(42)
 

models/antelopev2.zip

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