BestWishYsh commited on
Commit
7960eba
·
verified ·
1 Parent(s): 461e832

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -83,8 +83,8 @@ if not isinstance(eva_transform_std, (list, tuple)):
83
  eva_transform_mean = eva_transform_mean
84
  eva_transform_std = eva_transform_std
85
 
86
- face_main_model = FaceAnalysis(name='antelopev2', root=os.path.join(model_path, "face_encoder"), providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
87
- handler_ante = insightface.model_zoo.get_model(f'{model_path}/face_encoder/models/antelopev2/glintr100.onnx', providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
88
  face_main_model.prepare(ctx_id=0, det_size=(640, 640))
89
  handler_ante.prepare(ctx_id=0)
90
 
 
83
  eva_transform_mean = eva_transform_mean
84
  eva_transform_std = eva_transform_std
85
 
86
+ face_main_model = FaceAnalysis(name='antelopev2', root=os.path.join(model_path, "face_encoder"), providers=['CUDAExecutionProvider'])
87
+ handler_ante = insightface.model_zoo.get_model(f'{model_path}/face_encoder/models/antelopev2/glintr100.onnx', providers=['CUDAExecutionProvider'])
88
  face_main_model.prepare(ctx_id=0, det_size=(640, 640))
89
  handler_ante.prepare(ctx_id=0)
90