Jyothirmai commited on
Commit
1cc9796
·
verified ·
1 Parent(s): 45f9634

Update tester.py

Browse files
Files changed (1) hide show
  1. tester.py +1 -1
tester.py CHANGED
@@ -82,7 +82,7 @@ class CaptionSampler(object):
82
  self.word_model.eval()
83
 
84
  # imageData = Image.open(image_file).convert('RGB')
85
- imageData = self.transform(imageData)
86
  imageData = imageData.unsqueeze_(0)
87
 
88
  print(imageData.shape)
 
82
  self.word_model.eval()
83
 
84
  # imageData = Image.open(image_file).convert('RGB')
85
+ imageData = self.transform(image_file)
86
  imageData = imageData.unsqueeze_(0)
87
 
88
  print(imageData.shape)