Spaces:

Lokesh1024
/

Image_Cap

Build error

Lokesh1024 commited on Jul 20, 2024

Commit

743c05d

verified ·

1 Parent(s): 215e3fc

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -323,6 +323,6 @@ def get_caption_model():
     try:
         caption_model.load_weights('saved_models/image_captioning_coco_weights.h5')
     except FileNotFoundError:
-        caption_model.load_weights('Image_Cap/saved_models/image_captioning_coco_weights.h5')
     return caption_model

     try:
         caption_model.load_weights('saved_models/image_captioning_coco_weights.h5')
     except FileNotFoundError:
+        print("Not Found")
     return caption_model