Update app.py

app.py

@@ -6,6 +6,7 @@ import urllib.request
 import io
 from utils import *
 
+
 # Initialize labels and model
 labels = gen_labels()
 model = model_arc()  # Assuming this function initializes and returns a trained model
@@ -50,7 +51,7 @@ try:
     st.image(image, width = 256, caption = 'Uploaded Image')
     if st.button('Predict'):
         img = preprocess(image)
-
+        global img
         model = model_arc()
         #model.load_weights("classify_model.h5")
 
@@ -64,6 +65,14 @@ try:
         print(prediction)
         
         st.info('Hey! The uploaded image has been classified as " {} waste " '.format(labels[np.argmax(prediction[0], axis=-1)]))
+        
+        if img == 'paper' or 'cardboard' or 'metal' or 'glass':
+            return(" 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.')
+        
 except Exception as e:
   st.info(e)
   pass