tuan2308 commited on
Commit
17277c3
·
verified ·
1 Parent(s): 90c3c89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,9 +35,9 @@ def swap_faces(source_img, target_img):
35
 
36
  # Kiểm tra xem có phát hiện được khuôn mặt hay không
37
  if len(source_faces) == 0:
38
- return "Không tìm thấy khuôn mặt trong ảnh nguồn."
39
  if len(target_faces) == 0:
40
- return "Không tìm thấy khuôn mặt trong ảnh đích."
41
 
42
  # Chọn khuôn mặt đầu tiên trong ảnh nguồn
43
  source_face = source_faces[0]
 
35
 
36
  # Kiểm tra xem có phát hiện được khuôn mặt hay không
37
  if len(source_faces) == 0:
38
+ raise gr.Error("Không tìm thấy khuôn mặt trong ảnh nguồn.")
39
  if len(target_faces) == 0:
40
+ raise gr.Error("Không tìm thấy khuôn mặt trong ảnh đích.")
41
 
42
  # Chọn khuôn mặt đầu tiên trong ảnh nguồn
43
  source_face = source_faces[0]