lixiang46 commited on
Commit
211b362
β€’
1 Parent(s): 97c0886
Files changed (1) hide show
  1. app.py +1 -1
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