Spaces:

PKUWilliamYang
/

StyleGANEX

Runtime error

PKUWilliamYang commited on May 11, 2023

Commit

f740dc7

1 Parent(s): f58f170

Update webUI/styleganex_model.py

Files changed (1) hide show

webUI/styleganex_model.py CHANGED Viewed

@@ -32,7 +32,7 @@ class Model():
         self.task_name = None
         self.editing_w = None
         self.pspex = None
-        self.landmarkpredictor = dlib.shape_predictor(hf_hub_download('PKUWilliamYang/VToonify', 'models/shape_predictor_68_face_landmarks.dat'))
         self.transform = transforms.Compose([
             transforms.ToTensor(),
             transforms.Normalize(mean=[0.5, 0.5, 0.5],std=[0.5,0.5,0.5]),

         self.task_name = None
         self.editing_w = None
         self.pspex = None
+        self.landmarkpredictor = dlib.shape_predictor(hf_hub_download('PKUWilliamYang/StyleGANEX', 'pretrained_models/shape_predictor_68_face_landmarks.dat'))
         self.transform = transforms.Compose([
             transforms.ToTensor(),
             transforms.Normalize(mean=[0.5, 0.5, 0.5],std=[0.5,0.5,0.5]),