mirfan commited on
Commit
d2d6de0
·
1 Parent(s): 13a437f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -129,8 +129,8 @@ def display_UI():
129
  new_model = ImageCaptioningModel(cnn_model=cnn_model, encoder=encoder, decoder=decoder, image_aug=image_augmentation)
130
  # Call the model with some dummy input data to create its variables
131
  dummy_img = tf.zeros((1, 229, 229, 3))
132
- dummy_seq = tf.zeros((1, 5, 25))
133
- new_model(dummy_seq, dummy_img)
134
 
135
  # Load the weights
136
  new_model.load_weights('model_weights.h5')
 
129
  new_model = ImageCaptioningModel(cnn_model=cnn_model, encoder=encoder, decoder=decoder, image_aug=image_augmentation)
130
  # Call the model with some dummy input data to create its variables
131
  dummy_img = tf.zeros((1, 229, 229, 3))
132
+ dummy_seq = tf.zeros((1, 5, 512))
133
+ new_model((dummy_img, dummy_seq))
134
 
135
  # Load the weights
136
  new_model.load_weights('model_weights.h5')