Update app.py

app.py

@@ -33,6 +33,7 @@ import shutil
 import pandas as pd
 import numpy as np
 #import matplotlib.pyplot as plt
+from io import BytesIO
 import torch.nn.functional as F
 
 # Set the device (GPU or CPU)
@@ -173,7 +174,7 @@ enable = st.checkbox("Enable camera")
 picture_path = st.camera_input("Take a picture", disabled=not enable)
 if picture_path:
     # Create DataLoader for Webcam Image
-    webcam_ds = dataset_prod_obj.create_dataset(picture_path)
+    webcam_ds = dataset_prod_obj.create_dataset(BytesIO(picture_path))
     webcam_dl = DataLoader(webcam_ds, batch_size=BATCH_SIZE)
     
     # Run the predictions