Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,8 @@ from tensorflow.keras.models import load_model
|
|
11 |
model = load_model('model520.h5')
|
12 |
|
13 |
#prediction classes
|
14 |
-
classnames = ['paper', 'cardboard', 'plastic', 'metal', 'food', 'battery', 'shoes', 'clothes', 'glass', 'medical']
|
|
|
15 |
|
16 |
#prediction function
|
17 |
def predict_image(img):
|
|
|
11 |
model = load_model('model520.h5')
|
12 |
|
13 |
#prediction classes
|
14 |
+
#classnames = ['paper', 'cardboard', 'plastic', 'metal', 'food', 'battery', 'shoes', 'clothes', 'glass', 'medical']
|
15 |
+
classnames = ['battery','cardboard','clothes','food','glass','medical','metal','paper','plastic','shoes']
|
16 |
|
17 |
#prediction function
|
18 |
def predict_image(img):
|