Update app.py

app.py

@@ -79,11 +79,11 @@ word_to_index = StringLookup(
 def predict_caption(file):
     filename = Image.open(file)
     image = filename.convert('RGB')
+    image = np.array(image)
     gru_state = tf.zeros((1, ATTENTION_DIM))
-    
-    img = tf.image.decode_jpeg(np.array(image), channels=IMG_CHANNELS)
-    img = tf.image.resize(img, (IMG_HEIGHT, IMG_WIDTH))
-    img = img / 255
+
+    resize = tf.image.resize(image, (IMG_HEIGHT, IMG_WIDTH))
+    img = np.expand_dims(resize/255, 0)
     
     encoder = load_encoder_model()
     features = encoder(tf.expand_dims(img, axis=0))