Update app.py

app.py

@@ -64,7 +64,6 @@ def main():
             if os.path.isfile("known_face.jpg"):
                 known_image = face_recognition.load_image_file("known_face.jpg")
                 known_encoding = face_recognition.face_encodings(known_image)[0]
-                #known_encoding = face_recognition.face_encodings(known_image)
             else:
                 known_encoding = None
 
@@ -91,8 +90,7 @@ def main():
 
         if st.button("Register Known Face"):
             if image is not None:
-                cv2_img = cv2.imdecode(np.frombuffer(image.getvalue(), np.uint8), cv2.IMREAD_COLOR)
-                cv2.imwrite("known_face.jpg", cv2_img)
+                cv2.imwrite("known_face.jpg", np.array(image))
                 st.success("Known face registered successfully!")
 
         sidebar_html = "<div style='display:flex;flex-direction:column;'>"