Update app.py

app.py

@@ -18,17 +18,29 @@ Images     = []
 classnames = []
 
 # read images and train the face_recognition package
-img1 = cv2.imread(p1)
-Images.append(img1)
-classnames.append("Sarwan")
+# img1 = cv2.imread(p1)
+# Images.append(img1)
+# classnames.append("Sarwan")
 
-img2 = cv2.imread(p2)
-Images.append(img2)
-classnames.append("RattanTata")
+# img2 = cv2.imread(p2)
+# Images.append(img2)
+# classnames.append("RattanTata")
+
+# img3 = cv2.imread(p3)
+# Images.append(img3)
+# classnames.append("RavinderKaur")
+
+directory = "facerecognition"
+myList = os.listdir(directory)
+
+
+for cls in myList:
+    if os.path.splitext(cls)[1] in [".jpg", ".jpeg",".png"]:
+        img_path = os.path.join(directory, cls)
+        curImg = cv2.imread(img_path)
+        Images.append(curImg)
+        classnames.append(os.path.splitext(cls)[0])
 
-img3 = cv2.imread(p3)
-Images.append(img3)
-classnames.append("RavinderKaur")
 
 # Load images for face recognition
 encodeListknown = [face_recognition.face_encodings(img)[0] for img in Images]