Charbel Malo
commited on
Update face_swapper.py
Browse files- face_swapper.py +1 -1
face_swapper.py
CHANGED
@@ -42,7 +42,7 @@ def norm_crop2(img, landmark, image_size=112, mode='arcface'):
|
|
42 |
|
43 |
|
44 |
class Inswapper():
|
45 |
-
def __init__(self, model_file=None, batch_size=32, providers=['
|
46 |
self.model_file = model_file
|
47 |
self.batch_size = batch_size
|
48 |
|
|
|
42 |
|
43 |
|
44 |
class Inswapper():
|
45 |
+
def __init__(self, model_file=None, batch_size=32, providers=['CUDAExecutionProvider']):
|
46 |
self.model_file = model_file
|
47 |
self.batch_size = batch_size
|
48 |
|