Views
No views yet
1from transformers import TrOCRProcessor, VisionEncoderDecoderModel
2from PIL import Image
3
4model_name = "eldrar/cyrillic-large-handwritten-uk-rukopys"
5processor = TrOCRProcessor.from_pretrained(model_name)
6model = VisionEncoderDecoderModel.from_pretrained(model_name)
7
8image = Image.open("line.jpg")
9pixel_values = processor(image, return_tensors="pt").pixel_values
10ids = model.generate(pixel_values)
11print(processor.batch_decode(ids, skip_special_tokens=True)[0])