Update app.py

app.py

@@ -21,19 +21,17 @@ for img in source_imgs:
   source_faces.append(extract_faces(detector, img)[0])
 source_embeddings = get_embeddings(model, source_faces)
 
-index = faiss.IndexFlatL2(4096)
-index.add(np.array(source_embeddings))
-
-# set image
-image = 'group.jpg'
-
 def find_names(image):
-  imgs = extract_faces(detector, image)
+  imgs, _ = extract_faces(detector, image)
   embeds = get_embeddings(model, imgs)
-  D, I = index.search(np.array(embeds), 1)
+  d = np.zeros((len(source_embeddings), len(embeds)))
+  for i, s in enumerate(source_embeddings):
+    for j, t in enumerate(embeds):
+      d[i][j] = findCosineDistance(s, t)
+  ids = np.argmin(d, axis = 0)
   names = ""
-  for i in I:
-    names+=source_imgs[i[0]].split("/")[-1].split(".")[0]
+  for i in ids:
+    names+=source_imgs[i].split("/")[-1].split(".")[0]
     names+= ","
   return names