update

app.py

@@ -6,6 +6,7 @@ from utils import *
 cloth_examples = get_cloth_examples()
 pose_examples = get_pose_examples()
 tip1, tip2 = get_tips()
+face_detector = MTCNN()
 
 # Description
 title = r"""
@@ -32,6 +33,11 @@ def onUpload():
 def onClick(cloth_id, pose_image, pose_id, size, request: gr.Request):
     if pose_image is None:
         return None, "no pose image found !", ""
+
+    faces = face_detector.detect_faces(img)
+    if len(faces)==0:
+        return None, "Failed !!! No face detected !!! please upload a human photo!!! Not clothing photo!!!", ""
+        
     # pose_id, cloth_id = pose_id['label'], cloth_id['label']
     # print(pose_id, cloth_id, size, (pose_image is None), len(pose_id)>0)
     if len(pose_id)>0:

requirements.txt

@@ -3,4 +3,5 @@ numpy
 requests
 gradio==3.41.2
 gradio-client==0.5.0
-
+mtcnn
+tensorflow