Update tester.py

tester.py

@@ -81,12 +81,12 @@ class CaptionSampler(object):
         self.sentence_model.eval()
         self.word_model.eval()
         
-        # imageData = image_file.convert('RGB')
-        # imageData = self.transform(imageData)
-        # imageData = imageData.unsqueeze_(0)
+        imageData = image_file.convert('RGB')
+        imageData = self.transform(imageData)
+        imageData = imageData.unsqueeze_(0)
 
     
-        image = self.__to_var(image_file, requires_grad=False)
+        image = self.__to_var(imageData, requires_grad=False)
     
         visual_features, avg_features = self.extractor.forward(image)