Hugging Face's logo

Spaces:
mirfan
/
NLP-APP
Runtime error

App Files Community
1
mirfan commited on
Commit
2bb7076
·
1 Parent(s): f8f8514

init!

Browse files
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')
 
542
  cnn_model=cnn_model, encoder=encoder, decoder=decoder, image_aug=image_augmentation,
543
  )
544
 
545
+ new_model.load_weights('model_weights.h5')