Update app.py

app.py

@@ -15,6 +15,7 @@ labels = ['compost', 'e-waste', 'recycle', 'trash']
 
 
 def classify_image(input):
+    input = input.reshape((-1, 224, 224, 3))
     prediction = pre_trained_model.predict(input) 
     confidences = {labels[i]: float(prediction[i]) for i in range(4)}
     return confidences