Update app.py

app.py

@@ -5,11 +5,11 @@ import numpy as np
 # import face_recognition
 import subprocess
 import sys
-subprocess.check_call([sys.executable, "-m", "pip", "install", 'git+https://github.com/bit-guber/retinaface.git', "--force-reinstall"])
-
-from retinaface import RetinaFace
+# subprocess.check_call([sys.executable, "-m", "pip", "install", 'git+https://github.com/bit-guber/retinaface.git', "--force-reinstall"])
 
+# from retinaface import RetinaFace
 
+from deepface import DeepFace
 
 torch.backends.cudnn.benchmark = True
 
@@ -169,8 +169,9 @@ def image_inference(image):
     # 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 = RetinaFace.extract_faces( image, align = True)
+    # faces_mean = [ x.mean() for x in faces ]
+    faces = [ x['face'] for x in  DeepFace.extract_faces( image )]
     # 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 ]
     ages, genders, beards, blurs, ethncities, masks = AnalysisFeatures( faces )