Update app.py

app.py

@@ -8,7 +8,7 @@ model = load_model('DS11Sudhanva.h5')
 classnames = ['cardboard', 'metal','paper','plastic','trash','green-glass','white-glass','brown-glass','clothes','biological','battery','shoes']
 
 def predict(img):
-    img=img.reshape(-1,298, 384,3)
+    img=img.reshape(1,298, 384,3)
     prediction = model.predict(img[0])
     return {classnames[i]: float(prediction[i]) for i in range(len(classnames))}