Spaces:

fbrynpk
/

image-caption-generator

Runtime error

fbrynpk commited on Sep 26, 2023

Commit

7b08a51

•

1 Parent(s): 88111ff

Update app to use new model

Files changed (1) hide show

model.py CHANGED Viewed

@@ -315,8 +315,8 @@ def get_caption_model():
     caption_model.decoder(sample_y, sample_enc_out, training=False)
     try:
-        caption_model.load_weights('models/trained_coco_weights.h5')
     except FileNotFoundError:
-        caption_model.load_weights('image-caption-generator/models/trained_coco_weights.h5')
     return caption_model

     caption_model.decoder(sample_y, sample_enc_out, training=False)
     try:
+        caption_model.load_weights('models/trained_coco_weights_2.h5')
     except FileNotFoundError:
+        caption_model.load_weights('image-caption-generator/models/trained_coco_weights_2.h5')
     return caption_model