charbel-malo commited on
Commit
10ad106
·
verified ·
1 Parent(s): 02b8554

Update face_swapper.py

Browse files
Files changed (1) hide show
  1. 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