Update generator.py

generator.py

@@ -273,12 +273,14 @@ class RetroArtConverter:
             print("Detecting faces...")
             try:
                 image_np = cv2.cvtColor(np.array(resized_image), cv2.COLOR_RGB2BGR)
-                image_np = sorted(image_np, key=lambda x:(x['bbox'][2]-x['bbox'][0])*x['bbox'][3]-x['bbox'][1])[-1] # only use the maximum face
                 faces = self.face_app.get(image_np)
-                face = faces[0])
                 
                 if len(faces) > 0:
                     has_detected_faces = True
+                    face = faces[0]
+                    print(f"  [OK] Face detected (score: {face.det_score:.3f})")
+                    
+                    # Get face keypoints image
                     face_kps_image = draw_kps(resized_image, face.kps)
                     
                     # Get face embeddings (512D from InsightFace)