Spaces:

Jyothirmai
/

image-captioning-chest-xrays

Sleeping

Jyothirmai commited on Feb 27

Commit

df64df2

•

1 Parent(s): 7f1c94f

Update tester.py

Files changed (1) hide show

tester.py CHANGED Viewed

@@ -17,11 +17,11 @@ class CaptionSampler(object):
     def __init__(self):
         # Default configuration values
         self.args = {
-            "model_dir": "/Users/jkottu/Desktop/image-captioning-chest-xrays/vit-coatten",
-            "image_dir": "./data/images",
-            "caption_json": "data/new_data/captions.json",
-            "vocab_path": "/Users/jkottu/Desktop/image-captioning-chest-xrays/vit-coatten/vocab.pkl",
-            "file_lists": "data/new_data/test_data.txt",
             "load_model_path": "train_best_loss.pth.tar",
             "resize": 224,
             "cam_size": 224,

     def __init__(self):
         # Default configuration values
         self.args = {
+            "model_dir": "",
+            "image_dir": "",
+            "caption_json": "",
+            "vocab_path": "vocab.pkl",
+            "file_lists": "",
             "load_model_path": "train_best_loss.pth.tar",
             "resize": 224,
             "cam_size": 224,