test

handler.py

@@ -35,9 +35,16 @@ class EndpointHandler():
     def __init__(self, model_dir):
         
         print("Loading FaceAnalysis", model_dir)
-        self.app = FaceAnalysis(name="buffalo_l", providers=["CPUExecutionProvider"])
-        self.app.prepare(ctx_id=0, det_size=(640, 640))
+        # self.app = FaceAnalysis(name="buffalo_l", providers=["CPUExecutionProvider"])
+        # self.app.prepare(ctx_id=0, det_size=(640, 640))
 
+        self.app = FaceAnalysis(
+            name="antelopev2",
+            root="./antelopev2",
+            providers=["CPUExecutionProvider"],
+        )
+        self.app.prepare(ctx_id=0, det_size=(640, 640))
+        
         openpose = OpenposeDetector.from_pretrained("lllyasviel/ControlNet")
         # depth_anything = DepthAnything.from_pretrained('LiheYoung/depth_anything_vitl14').to(device).eval()