Lokesh1024 commited on
Commit
ae64456
·
verified ·
1 Parent(s): 778ad36

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -2
model.py CHANGED
@@ -14,7 +14,7 @@ UNITS = 512
14
 
15
 
16
  # LOADING DATA
17
- vocab = pickle.load(open('Image_Cap/saved_vocabulary/vocab_coco.file', 'rb'))
18
 
19
  tokenizer = tf.keras.layers.TextVectorization(
20
  # max_tokens=VOCABULARY_SIZE,
@@ -321,7 +321,7 @@ def get_caption_model():
321
  caption_model.decoder(sample_y, sample_enc_out, training=False)
322
 
323
  try:
324
- caption_model.load_weights('Image_Cap/saved_models/image_captioning_coco_weights.h5')
325
  except FileNotFoundError:
326
  caption_model.load_weights('Image-Cap/saved_models/image_captioning_coco_weights.h5')
327
 
 
14
 
15
 
16
  # LOADING DATA
17
+ vocab = pickle.load(open('saved_vocabulary/vocab_coco.file', 'rb'))
18
 
19
  tokenizer = tf.keras.layers.TextVectorization(
20
  # max_tokens=VOCABULARY_SIZE,
 
321
  caption_model.decoder(sample_y, sample_enc_out, training=False)
322
 
323
  try:
324
+ caption_model.load_weights('saved_models/image_captioning_coco_weights.h5')
325
  except FileNotFoundError:
326
  caption_model.load_weights('Image-Cap/saved_models/image_captioning_coco_weights.h5')
327