Spaces:
Sleeping
Sleeping
Commit
·
51ba127
1
Parent(s):
f0dfb12
Update app.py
Browse files
app.py
CHANGED
@@ -75,7 +75,7 @@ def generate_caption(image_name):
|
|
75 |
feature = vgg16_model.predict(image)
|
76 |
# predict the caption
|
77 |
y_pred = predict_caption(feature, max_length)
|
78 |
-
return y_pred.
|
79 |
|
80 |
|
81 |
st.title("""
|
|
|
75 |
feature = vgg16_model.predict(image)
|
76 |
# predict the caption
|
77 |
y_pred = predict_caption(feature, max_length)
|
78 |
+
return y_pred.replace("startseq", "").replace("endseq", "").strip()
|
79 |
|
80 |
|
81 |
st.title("""
|