Update app.py

app.py

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