Mr-Vicky-01 commited on
Commit
e55aa90
1 Parent(s): 7138b6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=35)
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')