Update app.py

app.py

@@ -165,10 +165,11 @@ def image_inference(image):
     # rawfaces = postProcessed( rawfaces, image.shape[:2] )
     
     # rawfaces = face_recognition.face_locations( image, number_of_times_to_upsample = 1 , model="hog")
-    rawfaces = []
-    for name, keys in RetinaFace.detect_faces( image ).items():
-        rawfaces.append( keys['facial_area']  )
-    faces = [ image[top:bottom, left:right].copy() for (top, left, bottom, right) in rawfaces ]
+    # rawfaces = []
+    # for name, keys in RetinaFace.detect_faces( image ).items():
+    #     rawfaces.append( keys['facial_area']  )
+    # faces = [ image[top:bottom, left:right].copy() for (top, left, bottom, right) in rawfaces ]
+    faces = RetinaFace.extract_faces( image, align = True)
     faces_mean = [ x.mean() for x in faces ]
     # faces = [ image[x:w+x, y:h+y].copy() for (x, y, w, h) in rawfaces ]
     faces = [ Image.fromarray(x, mode = 'RGB') for x in faces ]