Spaces:
Build error
Build error
Update model.py
Browse files
model.py
CHANGED
|
@@ -14,7 +14,7 @@ UNITS = 512
|
|
| 14 |
|
| 15 |
|
| 16 |
# LOADING DATA
|
| 17 |
-
vocab = pickle.load(open('
|
| 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('
|
| 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 |
|