Update app.py

app.py

@@ -11,7 +11,8 @@ from tensorflow.keras.models import load_model
 model = load_model('model520.h5')
 
 #prediction classes
-classnames = ['paper', 'cardboard', 'plastic', 'metal', 'food', 'battery', 'shoes', 'clothes', 'glass', 'medical']
+#classnames = ['paper', 'cardboard', 'plastic', 'metal', 'food', 'battery', 'shoes', 'clothes', 'glass', 'medical']
+classnames = ['battery','cardboard','clothes','food','glass','medical','metal','paper','plastic','shoes']
 
 #prediction function
 def predict_image(img):