Update app.py

app.py

@@ -105,7 +105,8 @@ cat_decoder.to(device)
 def run_alignment(image_path):
   import dlib
   from e4e.utils.alignment import align_face
-  predictor = dlib.shape_predictor("bankholdup/stylegan_petbreeder/shape_predictor_68_face_landmarks.dat")
+  dlib_path = hf_hub_download(repo_id="bankholdup/stylegan_petbreeder", filename="shape_predictor_68_face_landmarks.dat")
+  predictor = dlib.shape_predictor(dlib_path)
   aligned_image = align_face(filepath=image_path, predictor=predictor) 
   print("Aligned image has shape: {}".format(aligned_image.size))
   return aligned_image