PKUWilliamYang commited on
Commit
f740dc7
1 Parent(s): f58f170

Update webUI/styleganex_model.py

Browse files
Files changed (1) hide show
  1. webUI/styleganex_model.py +1 -1
webUI/styleganex_model.py CHANGED
@@ -32,7 +32,7 @@ class Model():
32
  self.task_name = None
33
  self.editing_w = None
34
  self.pspex = None
35
- self.landmarkpredictor = dlib.shape_predictor(hf_hub_download('PKUWilliamYang/VToonify', 'models/shape_predictor_68_face_landmarks.dat'))
36
  self.transform = transforms.Compose([
37
  transforms.ToTensor(),
38
  transforms.Normalize(mean=[0.5, 0.5, 0.5],std=[0.5,0.5,0.5]),
 
32
  self.task_name = None
33
  self.editing_w = None
34
  self.pspex = None
35
+ self.landmarkpredictor = dlib.shape_predictor(hf_hub_download('PKUWilliamYang/StyleGANEX', 'pretrained_models/shape_predictor_68_face_landmarks.dat'))
36
  self.transform = transforms.Compose([
37
  transforms.ToTensor(),
38
  transforms.Normalize(mean=[0.5, 0.5, 0.5],std=[0.5,0.5,0.5]),