diff --git a/gcp/main.py b/gcp/main.py index e12ddd1..adceb53 100644 --- a/gcp/main.py +++ b/gcp/main.py @@ -42,7 +42,7 @@ def predict(request): image = image/255 # normalize the image in 0 to 1 range - img_array = tf.expand_dims(img, 0) + img_array = tf.expand_dims(image, 0) predictions = model.predict(img_array) print("Predictions:",predictions)