Update vitGPT.py
Browse files
vitGPT.py
CHANGED
|
@@ -345,5 +345,5 @@ def generate_caption(image,max_tokens=50,temperature=0.9,deterministic=True):
|
|
| 345 |
print(caption)
|
| 346 |
return caption
|
| 347 |
|
| 348 |
-
image = "/Users/jkottu/Desktop/image-captioning-chest-xrays/sample_images/CXR191_IM-0591-1001.png"
|
| 349 |
-
generate_caption(image)
|
|
|
|
| 345 |
print(caption)
|
| 346 |
return caption
|
| 347 |
|
| 348 |
+
# image = "/Users/jkottu/Desktop/image-captioning-chest-xrays/sample_images/CXR191_IM-0591-1001.png"
|
| 349 |
+
# generate_caption(image)
|