Update model.py

model.py

@@ -360,8 +360,8 @@ def get_caption_model():
     caption_model.decoder(sample_y, sample_enc_out, training=False)
 
     try:
-        caption_model.load_weights('saved_models/my_checkpoint.ckpt')
+        caption_model.load_weights('saved_models/model.ckpt')
     except FileNotFoundError:
-        caption_model.load_weights('Image-Captioning/saved_models/my_checkpoint.ckpt')
+        caption_model.load_weights('Image-Captioning/saved_models/model.ckpt')
 
     return caption_model