wokring with buffalo_l

handler.py

@@ -141,15 +141,13 @@ class EndpointHandler():
             # "depth": get_depth_map,
         }
         
-        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"])
+        # 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="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)