Update app.py

app.py

@@ -18,7 +18,10 @@ for r, _, f in os.walk(os.getcwd() + "/images"):
 
 source_faces = []
 for img in source_imgs:
-  source_faces.append(extract_faces(detector, img)[-1])
+    try:
+      source_faces.append(extract_faces(detector, img)[-1])
+    except:
+      print(f"{img} not found, Skipping.")
 source_embeddings = get_embeddings(model, source_faces)
 
 def find_names(image):