Update app.py

app.py

@@ -46,6 +46,7 @@ BATCH_SIZE = 8
 # Set Paths
 data_path = 'employees'
 model_path = 'vit_pytorch_GPU_1.pt'
+webcam_path = 'captured_image.jpg'
 
 # Set Title
 st.title("Employee Attendance System")
@@ -171,10 +172,15 @@ prod_dl = DataLoader(prod_ds, batch_size=BATCH_SIZE)
 
 # Read image from Camera
 enable = st.checkbox("Enable camera")
-picture_path = st.camera_input("Take a picture", disabled=not enable)
-if picture_path:
+picture = st.camera_input("Take a picture", disabled=not enable)
+if picture is not None:
+    img_bytes = picture.getvalue()
+    img = PIL.Image.open(img_bytes)
+    img.save(webcam_path, "JPEG")
+    st.write('Image saved as:',webcam_path)
+
     # Create DataLoader for Webcam Image
-    webcam_ds = dataset_prod_obj.create_dataset(BytesIO(picture_path))
+    webcam_ds = dataset_prod_obj.create_dataset(webcam_path)
     webcam_dl = DataLoader(webcam_ds, batch_size=BATCH_SIZE)
     
     # Run the predictions