Spaces:
Sleeping
Sleeping
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
|