Update app.py

app.py

@@ -3,14 +3,12 @@ import streamlit as st
 from PIL import Image
 import urllib.request
 import io
-from utils import *
 import tensorflow as tf
 
 # Initialize labels and model
 labels = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']
 model = tf.keras.models.load_model('classify_model.h5')
 
-
 # Streamlit UI
 st.markdown('''
  <div style="padding-bottom: 20px; padding-top: 20px; padding-left: 5px; padding-right: 5px">
@@ -24,21 +22,18 @@ st.markdown('''
  </div>
 ''', unsafe_allow_html=True)
 
-
 opt = st.selectbox(
     "How do you want to upload the image for classification?",
     ("Please Select", "Upload image via link", "Upload image from device"),
 )
 
 image = None
-prediction = None
 
 if opt == 'Upload image from device':
   file = st.file_uploader('Select', type=['jpg', 'png', 'jpeg'])
   if file:
       image = preprocess_image(file)
     
-
 elif opt == 'Upload image via link':
   img_url = st.text_input('Enter the Image Address')
   if st.button('Submit'):
@@ -48,19 +43,14 @@ elif opt == 'Upload image via link':
     except ValueError:
       st.error("Please Enter a valid Image Address!")
 
-
-
 try:
   if image is not None:
     st.image(image, width=256, caption='Uploaded Image')
     if st.button('Predict'):
-
-      img = labels[np.argmax(prediction[0], axis=-1)]
+      prediction = model.predict(image[np.newaxis, ...])
 
       print("---------------img-array---------------------")
-      print(img[np.newaxis, ...])
-      prediction = model.predict(img[np.newaxis, ...])
-
+      print(image[np.newaxis, ...])
       print("------------summary------------------------")
       print(model.summary())
       print("------------------------------------")
@@ -74,12 +64,4 @@ try:
             " therefore your item is recyclable. Please refer to https://www.wm.com/us/en/drop-off-locations to find a drop-off location near you.")
         elif img == 'plastic':
           return (
-            ' therefore you item may have a chance of being recyclable. Since this model has yet to recognize types of plastics, please refer to https://www.bing.com/ck/a?!&&p=c1474e95017548dfJmltdHM9MTcwMzcyMTYwMCZpZ3VpZD0xNmNjOTFiOS1hMDgwLTY5MmItMzBmNi04MmE1YTE3ODY4NDImaW5zaWQ9NTIyMA&ptn=3&ver=2&hsh=3&fclid=16cc91b9-a080-692b-30f6-82a5a1786842&psq=what+type+of+plastic+can+be+recycled&u=a1aHR0cHM6Ly93d3cucGxhc3RpY3Nmb3JjaGFuZ2Uub3JnL2Jsb2cvd2hpY2gtcGxhc3RpYy1jYW4tYmUtcmVjeWNsZWQ&ntb=1 to check if this item can be recycled.')
-        else:
-          return ('Your item is not recyclable. Please discard it safely.')
-
-      st.info(message(labels[np.argmax(prediction[0], axis=-1)]))
-
-except Exception as e:
-  st.info(e)
-  pass
+            ' therefore your item may have a chance of being recyclable. Since this model has yet to recognize types of plastics, please refer to https://www.bing.com/ck/a?!&&p=c1474e95017548dfJmltdHM9MTcwMzcyMTYwMCZpZ3U9Jm91dHB1dD1zY2VuYXJlc2NyaWFsJmF1dGhvcmU9ISBBbGJ1bXN0ZXImY2xhc3Nlc19yZGM9MjAzNTMzODU5MDE5OTg0MyZjb25kaXRpb25fZHVyPSUyNm51bTttb2QlM0QxJTI2YXJ0aWQlM0QxJTI2aW5zaWQlM0Q1MjIwJTI2cHQlM0QzJTI2dmVyPSUyNnBvc3QlM0QzJTI2cmVxJTNEJTNEJTIyJTIyJTI2Z3JvdXBpZCUzRCUyMjB1cyUyMiUyNmRpc3BsYXklM0QlMjJ1cyUyMiUyNmxvY2FsZSUzRCUyMmVuLVVTJTIyJTI2cmVxdWVzdF9pZCUzRCUyMiUyMiUyNmF1ZHN0cmVhbV9pZCUzRCUyMiUyMiUyNm5vbWVfc3RyZWFtJTNEMSUyNmNvbnRlbnQlM0REJTIyJTIyJTI2c2hvd19hcmNoaXZlJmNtcCUzRDE2Y2M5MWYxLWFhMjAtNjkyYi0zMGI2LTgyYTVhMTc4Njg0MyUyMiUzRSUzQyUyRmJvYXJkLmxlYWd1ZXMlM0Z1JTNCaHR0cHMlM0ElMkYlMkZ3d3cucGxhc3RpY3Nmb3JjaGFuZ2Uub3JnJTJGJiUzRnJlZiUzRHYlMjZwYWdlX29yaWdpbmFsJTNEJTI2aWQlM0QlMjZjbGlkZXRhJTNEMzEwMTU1NTk5JTJGY29udGFjdCUzRmNsdXN0ZXIlM0QlMjZwYXN0JTNCJTJGJTNFJTI4JTIwJTIxJTIwJTIwJTIwJTIwJTIwJTIxJTI5JTIwJTIxJTIwJTIxJTIwJTIwJTIwJTIxJTIwJTIwJTIxJTIwJTIwJTIwJTIwJTIxJTIwJTIxJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTI