Update app.py

app.py

@@ -8,11 +8,11 @@ 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(298,384,3)
-    images_list = []
+    img=img.reshape(-1,298,384,3)
+    """images_list = []
     images_list.append(np.array(img))
-    x = np.asarray(images_list)
-    prediction = model.predict(x)
+    x = np.asarray(images_list)"""
+    prediction = model.predict(img)[0]
     return {classnames[i]: float(prediction[i]) for i in range(len(classnames))}
 
 image = gr.inputs.Image(shape=(298, 384))