Update app.py

app.py

@@ -45,23 +45,25 @@ elif opt == 'Upload image via link':
       st.error("Please Enter a valid Image Address!")
 
 
-if image is not None:
-    st.image(image, width=256, caption='Uploaded Image')
-if st.button('Predict'):
-  prediction = model.predict(image[np.newaxis, ...])
+try:
+    if image is not None:
+        st.image(image, width=256, caption='Uploaded Image')
+    if st.button('Predict'):
+      prediction = model.predict(image[np.newaxis, ...])
 
-  print("---------------img-array---------------------")
-  print(image[np.newaxis, ...])
-  print("------------summary------------------------")
-  print(model.summary())
-  print("------------------------------------")
-  print(prediction)
+      print("---------------img-array---------------------")
+      print(image[np.newaxis, ...])
+      print("------------summary------------------------")
+      print(model.summary())
+      print("------------------------------------")
+      print(prediction)
 
-  st.info('Hey! The uploaded image has been classified as " {} waste " '.format(labels[np.argmax(prediction[0], axis=-1)]))
-
-  def message(img):
-    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 your item may have a chance of being recyclable.")
+      st.info('Hey! The uploaded image has been classified as " {} waste " '.format(labels[np.argmax(prediction[0], axis=-1)]))
 
+      def message(img):
+        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 your item may have a chance of being recyclable.")
+except:
+    print(f"An error occurred: {e}")