Update app.py

app.py

@@ -18,6 +18,7 @@ from io import BytesIO
 import os
 from PIL import Image
 import gdown
+import face_recognition
 
 def download_gdrive_url():
   url = 'https://drive.google.com/u/0/uc?id=1PPO2MCttsmSqyB-vKh5C7SumwFKuhgyj&export=download'
@@ -47,17 +48,15 @@ def inpaint(p, init_image, mask_image=None, strength=0.75, guidance_scale=7.5, g
     return return_image
     
 def identify_face(user_image):
-  img = cv2.imread(user_image.name) # read the resized image in cv2
+  # img = cv2.imread(user_image.name) # read the resized image in cv2
+  img = face_recognition.load_image_file(user_image.name)
   print(img.shape)
-  gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) # convert to grayscale
-  download_gdrive_url() #download the haarcascade face recognition stuff
-  haar_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
-  faces_rect = haar_cascade.detectMultiScale(gray_img, scaleFactor=1.1, minNeighbors=9)
-  for (x, y, w, h) in faces_rect[:1]:
-    mask = np.zeros(img.shape[:2], dtype="uint8")
-    print(mask.shape)
-    cv2.rectangle(mask, (x, y), (x+w, y+h),  255, -1)
+  face_locations = face_recognition.face_locations(img)
+  for face_location in face_locations: 
+    top, right, bottom, left = face_location
+    mask = np.zeros(image.shape[:2], dtype="uint8")
     print(mask.shape)
+    cv2.rectangle(mask, (left, top), (right, bottom),  255, -1)
     inverted_image = cv2.bitwise_not(mask)
     return inverted_image