Update app.py

app.py

@@ -22,13 +22,14 @@ class FaceSwapper:
         self.image_b_origin = None
 
     def get_face_landmarks(self, image):
-        results = self.face_mesh.process(cv2.cvtColor(np.array(image), 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
+        with self.face_mesh as face_mesh:
+            results = face_mesh.process(cv2.cvtColor(np.array(image), 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):
         if image_a: