Update app.py

app.py

@@ -46,6 +46,7 @@ def predict_class(image):
     img = tf.image.resize(img, [input_shape[0], input_shape[1]])
     img = tf.expand_dims(img, axis=0)
     prediction = model.predict(img)
+    print(prediction)
     return prediction
 
 # UI Design