bankholdup commited on
Commit
117d66b
1 Parent(s): 478bf17

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -105,7 +105,7 @@ cat_decoder.to(device)
105
  def run_alignment(image_path):
106
  import dlib
107
  from e4e.utils.alignment import align_face
108
- predictor = dlib.shape_predictor("shape_predictor_68_face_landmarks.dat")
109
  aligned_image = align_face(filepath=image_path, predictor=predictor)
110
  print("Aligned image has shape: {}".format(aligned_image.size))
111
  return aligned_image
 
105
  def run_alignment(image_path):
106
  import dlib
107
  from e4e.utils.alignment import align_face
108
+ predictor = dlib.shape_predictor("bankholdup/stylegan_petbreeder/shape_predictor_68_face_landmarks.dat")
109
  aligned_image = align_face(filepath=image_path, predictor=predictor)
110
  print("Aligned image has shape: {}".format(aligned_image.size))
111
  return aligned_image