Update app.py

app.py

@@ -4,26 +4,37 @@ import cv2
 import face_recognition
 import streamlit as st
 
-p1 = "raavi.jpg"
-p2 = "jivan.jpeg"
-p3 = "sejal.jpeg"
+#p1 = "raavi.jpg"
+#p2 = "jivan.jpeg"
+# p3 = "sejal.jpeg"
 
 st.title("Face Recognition")
 Images = []
 classnames = []
 
 # Read images and train the face_recognition package
-img1 = cv2.imread(p1)
-Images.append(img1)
-classnames.append("Ravi")
+# img1 = cv2.imread(p1)
+# Images.append(img1)
+# classnames.append("Ravi")
 
-img2 = cv2.imread(p2)
-Images.append(img2)
-classnames.append("Sirwan sir")
+# img2 = cv2.imread(p2)
+# Images.append(img2)
+# classnames.append("Sirwan sir")
 
-img3 = cv2.imread(p3)
-Images.append(img3)
-classnames.append("sejal")
+# img3 = cv2.imread(p3)
+# Images.append(img3)
+# classnames.append("sejal")
+
+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])
 
 # Load images for face recognition
 encodeListknown = [face_recognition.face_encodings(img)[0] for img in Images]