Spaces:

Jyothirmai
/

image-captioning-chest-xrays

Sleeping

Jyothirmai commited on Mar 4, 2024

Commit

e013e74

verified ·

1 Parent(s): 8feeeb8

Update tester.py

Files changed (1) hide show

tester.py CHANGED Viewed

@@ -83,7 +83,7 @@ class CaptionSampler(object):
         # imageData = image_file.convert('RGB')
         # imageData = self.transform(imageData)
-        imageData = image_file.unsqueeze_(0)
         print(imageData.shape)

         # imageData = image_file.convert('RGB')
         # imageData = self.transform(imageData)
+        # imageData = imageData.unsqueeze_(0)
         print(imageData.shape)