Aryan-EcoClim commited on
Commit
72125ff
·
1 Parent(s): 5867cce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -51,7 +51,6 @@ try:
51
  st.image(image, width = 256, caption = 'Uploaded Image')
52
  if st.button('Predict'):
53
  img = preprocess(image)
54
- global img
55
  model = model_arc()
56
  #model.load_weights("classify_model.h5")
57
 
@@ -64,8 +63,9 @@ try:
64
  print("------------------------------------")
65
  print(prediction)
66
 
67
- st.info('Hey! The uploaded image has been classified as " {} waste " '.format(labels[np.argmax(prediction[0], axis=-1)]))
68
-
 
69
  if img == 'paper' or 'cardboard' or 'metal' or 'glass':
70
  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.")
71
  elif img == 'plastic':
 
51
  st.image(image, width = 256, caption = 'Uploaded Image')
52
  if st.button('Predict'):
53
  img = preprocess(image)
 
54
  model = model_arc()
55
  #model.load_weights("classify_model.h5")
56
 
 
63
  print("------------------------------------")
64
  print(prediction)
65
 
66
+ st.info('Hey! The uploaded image has been classified as " {} waste " '.format(labels[np.argmax(prediction[0], axis=-1)]) + )
67
+
68
+ def
69
  if img == 'paper' or 'cardboard' or 'metal' or 'glass':
70
  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.")
71
  elif img == 'plastic':