ResearcherXman commited on
Commit
0add3fc
1 Parent(s): d5449b8
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -198,7 +198,6 @@ def generate_image(face_image, pose_image, prompt, negative_prompt, style_name,
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'])
204
 
 
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_emb = face_info['embedding']
202
  face_kps = draw_kps(convert_from_cv2_to_image(face_image_cv2), face_info['kps'])
203