Update model.py

model.py

@@ -321,8 +321,8 @@ def get_caption_model():
     caption_model.decoder(sample_y, sample_enc_out, training=False)
 
     try:
-        caption_model.load_weights('saved_models/image_captioning_coco_weights.h5')
+        caption_model.load_weights('/saved_models/updated_model.h5')
     except FileNotFoundError:
-        caption_model.load_weights('Image-Captioning/saved_models/image_captioning_coco_weights.h5')
+        caption_model.load_weights('Image-Captioning/saved_models/updated_model.h5')
 
     return caption_model