Update app.py

app.py

@@ -24,7 +24,7 @@ from torch.utils.data import Dataset, DataLoader
 
 # Other Generic Libraries
 import torch
-import PIL
+from PIL import Image
 import os
 import streamlit as st
 import gc
@@ -135,7 +135,7 @@ class CreateDatasetProd():
         pixel_values = []
         for path in img_paths:
             # Read and process Images
-            img = PIL.Image.open(path)
+            img = Image.open(path)
             img = self.transform_prod(img)
 
             # Scaling the video to ML model's desired format
@@ -174,9 +174,8 @@ prod_dl = DataLoader(prod_ds, batch_size=BATCH_SIZE)
 enable = st.checkbox("Enable camera")
 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")
+    #img = Image.open(picture)
+    picture.save(webcam_path, "JPEG")
     st.write('Image saved as:',webcam_path)
 
     # Create DataLoader for Webcam Image