Spaces:
Sleeping
Sleeping
Mr-Vicky-01
commited on
Commit
•
e55aa90
1
Parent(s):
7138b6d
Update app.py
Browse files
app.py
CHANGED
@@ -96,7 +96,7 @@ def google_image_testing(inp):
|
|
96 |
# extract features
|
97 |
img_feature = pre_trained_model.predict(image, verbose=0)
|
98 |
# predict the caption
|
99 |
-
predicted = predict_caption(model, img_feature, tokenizer, max_length=
|
100 |
return predicted
|
101 |
|
102 |
demo = gr.Interface(fn=google_image_testing, inputs='image',outputs='text',title='Image Captioner')
|
|
|
96 |
# extract features
|
97 |
img_feature = pre_trained_model.predict(image, verbose=0)
|
98 |
# predict the caption
|
99 |
+
predicted = predict_caption(model, img_feature, tokenizer, max_length=74)
|
100 |
return predicted
|
101 |
|
102 |
demo = gr.Interface(fn=google_image_testing, inputs='image',outputs='text',title='Image Captioner')
|