Update app.py

app.py

@@ -2,6 +2,7 @@ import cv2
 import numpy as np
 import torch
 import streamlit as st
+from PIL import Image  # Import Image from PIL
 from transformers import AutoModelForImageClassification, AutoFeatureExtractor
 
 # Load AI Model from Hugging Face
@@ -40,7 +41,7 @@ def main():
     
     elif uploaded_image is not None:
         # If an image is uploaded, process it
-        image = Image.open(uploaded_image)
+        image = Image.open(uploaded_image)  # Use PIL.Image to open the image
         image = np.array(image)
         image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
         predictions = process_frame(image)
@@ -51,3 +52,4 @@ if __name__ == "__main__":
     main()
 
 
+