Spaces:

nishantguvvada
/

Image-Captioning

Sleeping

nishantguvvada commited on Dec 25, 2023

Commit

1d792f8

•

1 Parent(s): 31f7642

removed np.expand_dims

Files changed (1) hide show

app.py CHANGED Viewed

@@ -83,7 +83,7 @@ def predict_caption(file):
     gru_state = tf.zeros((1, ATTENTION_DIM))
     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))

     gru_state = tf.zeros((1, ATTENTION_DIM))
     resize = tf.image.resize(image, (IMG_HEIGHT, IMG_WIDTH))
+    img = resize/255
     encoder = load_encoder_model()
     features = encoder(tf.expand_dims(img, axis=0))