Jyothirmai
commited on
Update tester.py
Browse files
tester.py
CHANGED
@@ -273,11 +273,11 @@ class CaptionSampler(object):
|
|
273 |
|
274 |
|
275 |
|
276 |
-
def main(image):
|
277 |
-
|
278 |
-
|
279 |
-
|
280 |
-
|
281 |
-
|
282 |
-
|
283 |
|
|
|
273 |
|
274 |
|
275 |
|
276 |
+
def main(image):
|
277 |
+
sampler = CaptionSampler()
|
278 |
+
# image = 'sample_images/CXR195_IM-0618-1001.png'
|
279 |
+
caption = sampler.sample(image)
|
280 |
+
print(caption[0])
|
281 |
+
|
282 |
+
return caption[0]
|
283 |
|