Update

app.py

@@ -63,7 +63,7 @@ class App:
 
     def __init__(self, device: torch.device):
         self.device = device
-        self.face_detector = self._create_dlib_landmark_model()
+        self.landmark_model = self._create_dlib_landmark_model()
         self.encoder = self._load_encoder()
         self.transform = self._create_transform()
 
@@ -143,7 +143,7 @@ class App:
         return exstyles
 
     def detect_and_align_face(self, image) -> np.ndarray:
-        image = align_face(filepath=image.name, predictor=self.face_detector)
+        image = align_face(filepath=image.name, predictor=self.landmark_model)
         return image
 
     @staticmethod