Spaces:

Anandx05
/

image-captioning-mscoco

App Files Files Community

Anandx05 commited on Jun 7, 2023

Commit

876082b

•

1 Parent(s): 8955de2

Update model.py

Files changed (1) hide show

model.py +1 -1

model.py CHANGED Viewed

@@ -14,7 +14,7 @@ UNITS = 512
 # LOADING DATA
-vocab = pickle.load(open('saved_vocabulary/vocab_coco.file', 'rb'))
 tokenizer = tf.keras.layers.TextVectorization(
     # max_tokens=VOCABULARY_SIZE,

 # LOADING DATA
+vocab = pickle.load(open('saved_vocabulary/vocab_coco_1', 'rb'))
 tokenizer = tf.keras.layers.TextVectorization(
     # max_tokens=VOCABULARY_SIZE,