Michael Rey commited on
Commit
b68bda0
·
1 Parent(s): 43803c8

modified app

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -1
src/streamlit_app.py CHANGED
@@ -97,7 +97,8 @@ with tab2:
97
  uploaded_file = st.file_uploader("Upload an Image", type=["jpg", "png", "jpeg"])
98
 
99
  if uploaded_file is not None:
100
- image = Image.open(uploaded_file).convert("RGB")
 
101
  st.image(image, caption="Uploaded Image", width=300)
102
 
103
  input_tensor = preprocess_image(image)
 
97
  uploaded_file = st.file_uploader("Upload an Image", type=["jpg", "png", "jpeg"])
98
 
99
  if uploaded_file is not None:
100
+ raw_bytes = uploaded_file.read()
101
+ image = Image.open(io.BytesIO(raw_bytes)).convert("RGB")
102
  st.image(image, caption="Uploaded Image", width=300)
103
 
104
  input_tensor = preprocess_image(image)