ResearcherXman commited on
Commit
d5449b8
1 Parent(s): c7c986d
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -197,6 +197,7 @@ def generate_image(face_image, pose_image, prompt, negative_prompt, style_name,
197
  if len(face_info) == 0:
198
  raise gr.Error(f"Cannot find any face in the image! Please upload another person image")
199
 
 
200
  face_info = face_info[-1]
201
  face_emb = face_info['embedding']
202
  face_kps = draw_kps(convert_from_cv2_to_image(face_image_cv2), face_info['kps'])
 
197
  if len(face_info) == 0:
198
  raise gr.Error(f"Cannot find any face in the image! Please upload another person image")
199
 
200
+ face_info = sorted(face_info, key=lambda x:(x['bbox'][2]-x['bbox'][0])*x['bbox'][3]-x['bbox'][1])[-1] # only use the maximum face
201
  face_info = face_info[-1]
202
  face_emb = face_info['embedding']
203
  face_kps = draw_kps(convert_from_cv2_to_image(face_image_cv2), face_info['kps'])