Update app.py

app.py

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