Spaces:
Runtime error
Runtime error
lixiang46
commited on
Commit
β’
3e679cb
1
Parent(s):
98c6239
update
Browse files
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 |
|