Update app.py

app.py

@@ -44,26 +44,24 @@ elif opt == 'Upload image via link':
     except ValueError:
       st.error("Please Enter a valid Image Address!")
 
-try:
-  if image is not None:
+
+if image is not None:
     st.image(image, width=256, caption='Uploaded Image')
-    if st.button('Predict'):
-      prediction = model.predict(image[np.newaxis, ...])
+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)]))
+  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.")
+  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 ValueError:
-    print("Value Error")