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 |
|