init

predict.py

@@ -0,0 +1,19 @@
+import numpy as np
+from keras.utils import load_img, img_to_array
+from keras.applications.vgg16 import preprocess_input, decode_predictions
+
+def predict(model, path):
+    # receive image path as CLI argument
+    img = load_img(path ,color_mode='rgb', target_size=(224, 224))
+
+    # loaded image to np array for model to read
+    x = img_to_array(img)
+    x.shape
+    x = np.expand_dims(x, axis=0)
+
+    # process array and make predictions
+    x = preprocess_input(x)
+    features = model.predict(x)
+    p = decode_predictions(features)
+
+    return p