Spaces:
Runtime error
Runtime error
lixiang46
commited on
Commit
β’
211b362
1
Parent(s):
97c0886
update
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ pipe = StableDiffusionXLPipeline(
|
|
43 |
)
|
44 |
|
45 |
class FaceInfoGenerator():
|
46 |
-
def __init__(self, root_dir = "
|
47 |
self.app = FaceAnalysis(name = 'antelopev2', root = root_dir, providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
|
48 |
self.app.prepare(ctx_id = 0, det_size = (640, 640))
|
49 |
|
|
|
43 |
)
|
44 |
|
45 |
class FaceInfoGenerator():
|
46 |
+
def __init__(self, root_dir = "./.insightface/"):
|
47 |
self.app = FaceAnalysis(name = 'antelopev2', root = root_dir, providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
|
48 |
self.app.prepare(ctx_id = 0, det_size = (640, 640))
|
49 |
|