File size: 223 Bytes
6eff0e6 | 1 2 3 4 5 6 7 8 | import model
import datasets
from PIL import Image
vlm = model.ImageNetCaptionModel.load_from_checkpoint('checkpoint-epoch=01-loss=0.13.ckpt')
image = Image.open("test_image_5.png")
print(vlm.generate(image=image))
|