ishworrsubedii commited on
Commit
ede97cc
·
verified ·
1 Parent(s): 6888585

Update src/components/faceswap.py

Browse files
Files changed (1) hide show
  1. src/components/faceswap.py +1 -1
src/components/faceswap.py CHANGED
@@ -12,7 +12,7 @@ warnings.filterwarnings("ignore", category=FutureWarning)
12
 
13
 
14
  class FaceSwapper:
15
- def __init__(self, app_name='buffalo_l', det_size=(640, 640), device="cuda"):
16
  model_dir = "/mannequin_to_model/.insightface/models"
17
  os.makedirs(model_dir, exist_ok=True)
18
  self.device = device
 
12
 
13
 
14
  class FaceSwapper:
15
+ def __init__(self, app_name='buffalo_l', det_size=(640, 640), device="cpu"):
16
  model_dir = "/mannequin_to_model/.insightface/models"
17
  os.makedirs(model_dir, exist_ok=True)
18
  self.device = device