Charles-Elena
/

InstantID

Model card Files Files and versions

yamildiego commited on Apr 4, 2024

Commit

5d7f3e0

·

1 Parent(s): bfbd999

antelov

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -53,8 +53,8 @@ class EndpointHandler():
         dir_path = os.path.join("", "models", "antelopev2")
         print(dir_path)
         print(model_dir)
-        self.app = FaceAnalysis(name="antelopev2", providers=["CPUExecutionProvider"])
         self.app.prepare(ctx_id=0, det_size=(640, 640))
         openpose = OpenposeDetector.from_pretrained("lllyasviel/ControlNet")

         dir_path = os.path.join("", "models", "antelopev2")
         print(dir_path)
         print(model_dir)
+        self.app = FaceAnalysis(name='antelopev2', root='./', providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
         self.app.prepare(ctx_id=0, det_size=(640, 640))
         openpose = OpenposeDetector.from_pretrained("lllyasviel/ControlNet")