Update app.py

app.py

@@ -7,6 +7,7 @@
 # from streamlit_webrtc import webrtc_streamer, VideoTransformerBase, RTCConfiguration, VideoProcessorBase, WebRtcMode
 import numpy as np
 import tensorflow
+from PIL import Image
 import cv2
 import streamlit as st
 from tensorflow import keras
@@ -117,10 +118,8 @@ def main():
         st.header("Image Prediction App")
         uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
         if uploaded_file is not None:
-            with st.proto.TempFile(suffix=".jpg") as temp:
-                temp.write(uploaded_file.getvalue())
-                temp.seek(0)
-                image = cv2.imread(temp.name)
+            image = np.array(Image.open(uploaded_file))
+            
             prediction = generate_prediction(image)
             st.image(prediction, use_column_width=True)
     # elif choice == "By Images":