Spaces:
Running
Running
added one more weights
Browse files
model.py
CHANGED
@@ -321,8 +321,8 @@ def get_caption_model():
|
|
321 |
caption_model.decoder(sample_y, sample_enc_out, training=False)
|
322 |
|
323 |
try:
|
324 |
-
caption_model.load_weights('saved_models/
|
325 |
except FileNotFoundError:
|
326 |
-
caption_model.load_weights('Image-Captioning/saved_models/
|
327 |
|
328 |
return caption_model
|
|
|
321 |
caption_model.decoder(sample_y, sample_enc_out, training=False)
|
322 |
|
323 |
try:
|
324 |
+
caption_model.load_weights('saved_models/image_captioning_transformer_weights_3.h5')
|
325 |
except FileNotFoundError:
|
326 |
+
caption_model.load_weights('Image-Captioning/saved_models/image_captioning_transformer_weights_3.h5')
|
327 |
|
328 |
return caption_model
|
saved_models/image_captioning_transformer_weights_3.h5
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:61d25373a21ccddf8bf24f67b2a1bb2037d565f8e88c85e20e7f5ef8527a90d9
|
3 |
+
size 247484996
|