Update app.py

app.py

@@ -158,7 +158,7 @@ with st.form("api_form"):
         # To read file as bytes:
         bytes_data = uploaded_file.getvalue()
         #st.write(bytes_data)
-        st.image(uploaded_file)
+        image = st.image(uploaded_file)
 
 
     submitted = st.form_submit_button("Submit")
@@ -169,7 +169,7 @@ with st.form("api_form"):
         #prediction = output[0]['label']
         #st.write(f'prediction = {prediction}')
         #st.text(output)
-        prediction = predict(bytes_data)
+        prediction = predict(image)
         st.write(prediction)