File size: 294 Bytes
5fa1a76
 
 
 
 
 
 
1
2
3
4
5
6
7
python
generated_ids = model.generate(pixel_values=pixel_values, max_length=50)
generated_caption = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
print(generated_caption)

a drawing of a pink and blue pokemon
Looks like the fine-tuned model generated a pretty good caption!