cbensimon HF staff commited on
Commit
ca87158
1 Parent(s): 50f3c39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ base_model = "frankjoshua/albedobaseXL_v13"
51
  patch_onnx_runtime()
52
 
53
  snapshot_download("okaris/antelopev2", local_dir="./models/antelopev2")
54
- face_analysis = FaceAnalysis(name='antelopev2', root='./', providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
55
  face_analysis.prepare(ctx_id=0, det_size=(640, 640))
56
 
57
  dtype = torch.float16
 
51
  patch_onnx_runtime()
52
 
53
  snapshot_download("okaris/antelopev2", local_dir="./models/antelopev2")
54
+ face_analysis = FaceAnalysis(name='antelopev2', root='./', providers=['CPUExecutionProvider'])
55
  face_analysis.prepare(ctx_id=0, det_size=(640, 640))
56
 
57
  dtype = torch.float16