Update face_swapper.py
Browse files- face_swapper.py +0 -3
face_swapper.py
CHANGED
@@ -46,7 +46,6 @@ def norm_crop2(img, landmark, image_size=112, mode='arcface'):
|
|
46 |
|
47 |
class Inswapper():
|
48 |
|
49 |
-
@spaces.GPU(enable_queue=True)
|
50 |
def __init__(self, model_file=None, batch_size=32, providers=['CUDAExecutionProvider,CPUExecutionProvider']):
|
51 |
self.model_file = model_file
|
52 |
self.batch_size = batch_size
|
@@ -58,7 +57,6 @@ class Inswapper():
|
|
58 |
self.session_options = onnxruntime.SessionOptions()
|
59 |
self.session = onnxruntime.InferenceSession(self.model_file, sess_options=self.session_options, providers=providers)
|
60 |
|
61 |
-
@spaces.GPU(enable_queue=True)
|
62 |
def forward(self, imgs, latents):
|
63 |
preds = []
|
64 |
for img, latent in zip(imgs, latents):
|
@@ -66,7 +64,6 @@ class Inswapper():
|
|
66 |
pred = self.session.run(['output'], {'target': img, 'source': latent})[0]
|
67 |
preds.append(pred)
|
68 |
|
69 |
-
@spaces.GPU(enable_queue=True)
|
70 |
def get(self, imgs, target_faces, source_faces):
|
71 |
imgs = list(imgs)
|
72 |
|
|
|
46 |
|
47 |
class Inswapper():
|
48 |
|
|
|
49 |
def __init__(self, model_file=None, batch_size=32, providers=['CUDAExecutionProvider,CPUExecutionProvider']):
|
50 |
self.model_file = model_file
|
51 |
self.batch_size = batch_size
|
|
|
57 |
self.session_options = onnxruntime.SessionOptions()
|
58 |
self.session = onnxruntime.InferenceSession(self.model_file, sess_options=self.session_options, providers=providers)
|
59 |
|
|
|
60 |
def forward(self, imgs, latents):
|
61 |
preds = []
|
62 |
for img, latent in zip(imgs, latents):
|
|
|
64 |
pred = self.session.run(['output'], {'target': img, 'source': latent})[0]
|
65 |
preds.append(pred)
|
66 |
|
|
|
67 |
def get(self, imgs, target_faces, source_faces):
|
68 |
imgs = list(imgs)
|
69 |
|