Update app.py

app.py

@@ -11,6 +11,7 @@ import pathlib
 from fastai.vision.all import *
 from fastai.imports import *
 from tensorflow.keras.models import model_from_json
+from mtcnn.mtcnn import MTCNN
 
 json_file = open('model.json', 'r')
 loaded_model_json = json_file.read()
@@ -76,11 +77,11 @@ def who_is_it(image):
         return ("Not in database")
 
 label = gr.outputs.Label()
-mtcnn = MTCNN(image_size=224)
+faceModel=MTCNN()
 def recog(image):
-    face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')
-    faces = face_cascade.detectMultiScale(image, scaleFactor=1.3, minNeighbors=5)
-    for x,y,w,h in faces:
+    faces = faceModel.detect_faces(image)
+    for face in faces:
+        x,y,w,h = face["box"]
         img = image[y:y+h, x:x+w]
     if face is not None:
         identity=who_is_it(img)