mirfan commited on
Commit
f8f8514
1 Parent(s): 1fe45e9
Files changed (1) hide show
  1. models.py +1 -1
models.py CHANGED
@@ -542,4 +542,4 @@ new_model = ImageCaptioningModel(
542
  cnn_model=cnn_model, encoder=encoder, decoder=decoder, image_aug=image_augmentation,
543
  )
544
 
545
- new_model.load_weights('model_weights.h5')
 
542
  cnn_model=cnn_model, encoder=encoder, decoder=decoder, image_aug=image_augmentation,
543
  )
544
 
545
+ new_model.load_weights('model_weights')