Update util.py

util.py

@@ -12,6 +12,9 @@ import scipy
 import scipy.ndimage
 import torchvision.transforms as transforms
 
+shape_predictor_path = hf_hub_download(repo_id="akhaliq/jojogan_dlib", filename="shape_predictor_68_face_landmarks.dat")
+
+
 google_drive_paths = {
     "models/stylegan2-ffhq-config-f.pt": "https://drive.google.com/uc?id=1Yr7KuD959btpmcKGAUsbAk5rPjX2MytK",
     "models/dlibshape_predictor_68_face_landmarks.dat": "https://drive.google.com/uc?id=11BDmNKS1zxSZxkgsEvQoKgFd8J264jKp",
@@ -128,7 +131,7 @@ def align_face(filepath, output_size=256, transform_size=1024, enable_padding=Tr
     :param filepath: str
     :return: PIL Image
     """
-    predictor = dlib.shape_predictor("dlibshape_predictor_68_face_landmarks.dat")
+    predictor = dlib.shape_predictor(shape_predictor_path)
     lm = get_landmark(filepath, predictor)
 
     lm_chin = lm[0: 17]  # left-right