Update model.py

model.py

@@ -14,7 +14,7 @@ UNITS = 512
 
 
 # LOADING DATA
-vocab = pickle.load(open('saved_vocabulary/vocab_coco_1', 'rb'))
+vocab = pickle.load(open('vocabulary/vocab_coco_1', 'rb'))
 
 tokenizer = tf.keras.layers.TextVectorization(
     # max_tokens=VOCABULARY_SIZE,
@@ -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('model weights/image_captioning_coco_weights.h5')
     except FileNotFoundError:
-        caption_model.load_weights('Image-Captioning/saved_models/image_captioning_coco_weights.h5')
+        caption_model.load_weights('image-captioning-mscoco/model weights/image_captioning_coco_weights.h5')
 
     return caption_model