Spaces:

Jyothirmai
/

image-captioning-chest-xrays

Runtime error

Jyothirmai commited on Mar 5, 2024

Commit

a77dcc0

verified ·

1 Parent(s): 0b8acb0

Update ViTCoAtt.py

Files changed (1) hide show

ViTCoAtt.py CHANGED Viewed

@@ -18,7 +18,7 @@ class CaptionSampler(object):
     def __init__(self):
         # Default configuration values
         self.args = {
-            "model_dir": "",
             "image_dir": "",
             "caption_json": "",
             "vocab_path": "vocab.pkl",

     def __init__(self):
         # Default configuration values
         self.args = {
+            "model_dir": "model/",
             "image_dir": "",
             "caption_json": "",
             "vocab_path": "vocab.pkl",