FoivosPar commited on
Commit
06c6140
1 Parent(s): 4d01ceb

initial demo

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -38,10 +38,7 @@ hf_hub_download(repo_id="FoivosPar/Arc2Face", filename="encoder/pytorch_model.bi
38
  hf_hub_download(repo_id="FoivosPar/Arc2Face", filename="arcface.onnx", local_dir="./models/antelopev2")
39
 
40
  # Load face detection and recognition package
41
- if device=="cuda":
42
- app = FaceAnalysis(name='antelopev2', root='./', providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
43
- else:
44
- app = FaceAnalysis(name='antelopev2', root='./', providers=['CPUExecutionProvider'])
45
  app.prepare(ctx_id=0, det_size=(640, 640))
46
 
47
  # Load pipeline
 
38
  hf_hub_download(repo_id="FoivosPar/Arc2Face", filename="arcface.onnx", local_dir="./models/antelopev2")
39
 
40
  # Load face detection and recognition package
41
+ app = FaceAnalysis(name='antelopev2', root='./', providers=['CPUExecutionProvider'])
 
 
 
42
  app.prepare(ctx_id=0, det_size=(640, 640))
43
 
44
  # Load pipeline