Spaces:
Sleeping
Sleeping
Commit
·
f0dfb12
1
Parent(s):
c49a07f
Update app.py
Browse files
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
|
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
|