saichandrapandraju commited on
Commit
f0dfb12
1 Parent(s): c49a07f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ def predict_caption(image, max_length):
50
  # get index with high probability
51
  yhat = np.argmax(yhat)
52
  # convert index to word
53
- word = idx_to_word(yhat, lstm_tokenizer)
54
  # stop if word not found
55
  if word is None:
56
  break
 
50
  # get index with high probability
51
  yhat = np.argmax(yhat)
52
  # convert index to word
53
+ word = idx_to_word(yhat)
54
  # stop if word not found
55
  if word is None:
56
  break