antelopev2

handler.py

@@ -35,18 +35,18 @@ class EndpointHandler():
         
         print("Loading FaceAnalysis", model_dir)
 
-        # self.app = FaceAnalysis(
-        #     name="antelopev2",
-        #     root=f"./antelopev2",
-        #     providers=["CPUExecutionProvider"],
-        # )
-        
         self.app = FaceAnalysis(
-        name="buffalo_l",
-        root="./",
-        providers=["CPUExecutionProvider"],
+            name="antelopev2",
+            root=f"./antelopev2",
+            providers=["CPUExecutionProvider"],
         )
         
+        # self.app = FaceAnalysis(
+        # name="buffalo_l",
+        # root="./",
+        # providers=["CPUExecutionProvider"],
+        # )
+        
         self.app.prepare(ctx_id=0, det_size=(640, 640))
         
         openpose = OpenposeDetector.from_pretrained("lllyasviel/ControlNet")