deal with mask

app.py

@@ -88,10 +88,12 @@ def face_swap_app(target_img, face_img):
     target_img = target_img.convert("RGB")
     face_img = face_img.convert("RGB")
 
+    edit_mask = create_face_mask(face_img)
+
     output_img, edit_image, change_image, mask, seed = ace_infer(
         reference_image=target_img,
         edit_image=face_img,
-        edit_mask=None,
+        edit_mask=edit_mask,
         prompt="Face swap",
         output_height=1024,
         output_width=1024,