changes

.gitattributes

@@ -29,3 +29,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.file filter=lfs diff=lfs merge=lfs -text
+vocab_2.file filter=lfs diff=lfs merge=lfs -text

model.py

@@ -6,7 +6,7 @@ import numpy as np
 
 # CONTANTS
 MAX_LENGTH = 40
-VOCABULARY_SIZE = 10000
+# VOCABULARY_SIZE = 10000
 BATCH_SIZE = 32
 BUFFER_SIZE = 1000
 EMBEDDING_DIM = 512
@@ -14,10 +14,10 @@ UNITS = 512
 
 
 # LOADING DATA
-vocab = pickle.load(open('saved_models/vocab.file', 'rb'))
+vocab = pickle.load(open('saved_vocabulary/vocab_1.file', 'rb'))
 
 tokenizer = tf.keras.layers.TextVectorization(
-    max_tokens=VOCABULARY_SIZE,
+    # max_tokens=VOCABULARY_SIZE,
     standardize=None,
     output_sequence_length=MAX_LENGTH,
     vocabulary=vocab
@@ -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_transformer_weights_3.h5')
+        caption_model.load_weights('saved_models/image_captioning_transformer_weights_2.h5')
     except FileNotFoundError:
-        caption_model.load_weights('Image-Captioning/saved_models/image_captioning_transformer_weights_3.h5')
+        caption_model.load_weights('Image-Captioning/saved_models/image_captioning_transformer_weights_2.h5')
 
     return caption_model

saved_vocabulary/vocab_1.file

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4df22d5dd84bbb3421dfce39b4affd49ba2dc08739627ee9cf036ae17d326d28
+size 860291

saved_vocabulary/vocab_2.file

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5397f3824177102eff3f2fe01eb8b6fffbe97671a8e8b102e20bd5e8e2064b7e
+size 1985473