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! |