Update app.py

app.py

@@ -17,8 +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 = tf.keras.applications.mobilenet_v2.predict(inp).flatten()
-    #prediction = pre_trained_model.predict(inp).flatten()
+    prediction = pre_trained_model.predict(inp).flatten()
     confidences = {labels[i]: float(prediction[i]) for i in range(4)}
     return confidences