first commit

model.py

@@ -48,11 +48,11 @@ class Model:
 
     @staticmethod
     def _create_dlib_landmark_model():
-        path = huggingface_hub.hf_hub_download(repo_id="akhaliq/jojogan_dlib", filename="shape_predictor_68_face_landmarks.dat"           )
+        path = huggingface_hub.hf_hub_download(repo_id="aijack/jojogan", filename="face_landmarks.dat"           )
         return dlib.shape_predictor(path)
 
     def _load_e4e(self) -> nn.Module:
-        ckpt_path = huggingface_hub.hf_hub_download(repo_id="akhaliq/JoJoGAN_e4e_ffhq_encode", filename="e4e_ffhq_encode.pt"    )
+        ckpt_path = huggingface_hub.hf_hub_download(repo_id="aijack/e4e", filename="e4e.pt"    )
         ckpt = torch.load(ckpt_path, map_location='cpu')
         opts = ckpt['opts']
         opts['device'] = self.device.type