Update app.py

app.py

@@ -78,9 +78,10 @@ word_to_index = StringLookup(
 ## Probabilistic prediction using the trained model
 def predict_caption(file):
     filename = Image.open(file)
+    image = filename.convert('RGB')
     gru_state = tf.zeros((1, ATTENTION_DIM))
     
-    img = tf.image.decode_jpeg(filename, channels=IMG_CHANNELS)
+    img = tf.image.decode_jpeg(np.array(image), channels=IMG_CHANNELS)
     img = tf.image.resize(img, (IMG_HEIGHT, IMG_WIDTH))
     img = img / 255