bufalo_1

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")