Update app.py

app.py

@@ -18,11 +18,11 @@ for r, _, f in os.walk(os.getcwd() + "/images"):
 
 source_faces = []
 for img in source_imgs:
-  source_faces.append(extract_faces(detector, img)[0])
+  source_faces.append(extract_faces(detector, img)[-1])
 source_embeddings = get_embeddings(model, source_faces)
 
 def find_names(image):
-  imgs, _ = extract_faces(detector, image)
+  imgs = extract_faces(detector, image)
   embeds = get_embeddings(model, imgs)
   d = np.zeros((len(source_embeddings), len(embeds)))
   for i, s in enumerate(source_embeddings):