advcloud
commited on
Commit
·
0d0de39
1
Parent(s):
d855d82
first commit
Browse files
model.py
CHANGED
@@ -92,7 +92,7 @@ class Model:
|
|
92 |
return transform
|
93 |
|
94 |
def detect_and_align_face(self, image) -> PIL.Image.Image:
|
95 |
-
image = align_face(filepath=image
|
96 |
return image
|
97 |
|
98 |
@staticmethod
|
|
|
92 |
return transform
|
93 |
|
94 |
def detect_and_align_face(self, image) -> PIL.Image.Image:
|
95 |
+
image = align_face(filepath=image, predictor=self.landmark_model)
|
96 |
return image
|
97 |
|
98 |
@staticmethod
|