Spaces:
Runtime error
Runtime error
bankholdup
commited on
Commit
•
117d66b
1
Parent(s):
478bf17
Update app.py
Browse files
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
|