Spaces:

Jyothirmai
/

image-captioning-chest-xrays

Sleeping

Jyothirmai commited on Feb 27

Commit

9d79825

•

1 Parent(s): df64df2

Update tester.py

Files changed (1) hide show

tester.py CHANGED Viewed

@@ -162,7 +162,7 @@ class CaptionSampler(object):
         return ' '.join(sampled_caption)
     def __init_vocab(self):
-        with open(self.args["vocab_path"], 'rb') as f:
             vocab = pickle.load(f)
         return vocab

         return ' '.join(sampled_caption)
     def __init_vocab(self):
+        with open('vocab.pkl', 'rb') as f:
             vocab = pickle.load(f)
         return vocab