Spaces:
Running
Running
Commit
·
72125ff
1
Parent(s):
5867cce
Update app.py
Browse files
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':
|