Spaces:

Jyothirmai
/

image-captioning-chest-xrays

Sleeping

Jyothirmai commited on Mar 4, 2024

Commit

75462dc

verified ·

1 Parent(s): dadc353

Update tester.py

Files changed (1) hide show

tester.py CHANGED Viewed

@@ -86,7 +86,7 @@ class CaptionSampler(object):
         # imageData = imageData.unsqueeze_(0)
-        image = self.__to_var(imageData, requires_grad=False)
         visual_features, avg_features = self.extractor.forward(image)

         # imageData = imageData.unsqueeze_(0)
+        image = self.__to_var(image_file, requires_grad=False)
         visual_features, avg_features = self.extractor.forward(image)