lixiang46 commited on
Commit
3e679cb
β€’
1 Parent(s): 98c6239
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -44,12 +44,10 @@ pipe = StableDiffusionXLPipeline(
44
  )
45
 
46
  class FaceInfoGenerator():
47
- @spaces.GPU
48
  def __init__(self, root_dir = "./.insightface/"):
49
  self.app = FaceAnalysis(name = 'antelopev2', root = root_dir, providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
50
  self.app.prepare(ctx_id = 0, det_size = (640, 640))
51
 
52
- @spaces.GPU
53
  def get_faceinfo_one_img(self, face_image):
54
  face_info = self.app.get(cv2.cvtColor(np.array(face_image), cv2.COLOR_RGB2BGR))
55
 
 
44
  )
45
 
46
  class FaceInfoGenerator():
 
47
  def __init__(self, root_dir = "./.insightface/"):
48
  self.app = FaceAnalysis(name = 'antelopev2', root = root_dir, providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
49
  self.app.prepare(ctx_id = 0, det_size = (640, 640))
50
 
 
51
  def get_faceinfo_one_img(self, face_image):
52
  face_info = self.app.get(cv2.cvtColor(np.array(face_image), cv2.COLOR_RGB2BGR))
53