new try loading antelopev2

handler.py

@@ -141,9 +141,15 @@ class EndpointHandler():
             # "depth": get_depth_map,
         }
         
-        self.app = FaceAnalysis(name="buffalo_l", root="./", providers=["CPUExecutionProvider"])
+        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)) 
+
+        # Use ctx_id=0 for GPU        
+        # 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)