bug fiexd

pages/2_Face Detection and Creating Database.py

@@ -124,16 +124,22 @@ def step2_page():
                     pil_image.save(img_path)
                     st.success("Face added successfully!")
 
-    st.write("Now lets see your known-faces database!")
+
+    st.info("Now lets see your known-faces database!")
     images = os.listdir(img_dir)
     if len(images) <= 0:
-        st.error("No faces have been added yet.")
+        st.write("No faces have been added yet.")
     else:
         cols = st.columns(len(images))
-        for i, img in enumerate(images):
-            face_name = img.split("_")[1]
-            cols[i].image(os.path.join(img_dir, img), use_column_width="auto")
-            cols[i].write(face_name)
+        for i in range(len(images)):
+            col = cols[i]
+
+            # display faces
+            with col:
+                face = face_recognition.load_image_file(images[i])
+                face_name = images[i].split("_")[1]
+                st.image(face, use_column_width="auto")
+                st.write(face_name)
 
     if st.button("Clear All"):
         for img in images: