Update app.py

app.py

@@ -22,13 +22,17 @@ class FaceSwapper:
         self.image_b_origin = None
 
     def get_face_landmarks(self, image):
-        with self.face_mesh as face_mesh:
-            results = face_mesh.process(cv2.cvtColor(np.array(image), cv2.COLOR_BGR2RGB))
+        with self.mp_face_mesh.FaceMesh(static_image_mode=True, max_num_faces=1, refine_landmarks=True, min_detection_confidence=0.5) as face_mesh:
+            image_np = np.array(image)
+            results = face_mesh.process(cv2.cvtColor(image_np, cv2.COLOR_BGR2RGB))
+
             if not results.multi_face_landmarks:
                 return None
+
             landmarks = results.multi_face_landmarks[0].landmark
             w, h = image.size
             face_landmarks = [(int(p.x * w), int(p.y * h)) for p in landmarks]
+
             return face_landmarks
 
     def load_images(self, image_a, image_b):