Views
No views yet
1from transformers import TrOCRProcessor, VisionEncoderDecoderModel
2from PIL import Image
3
4processor = TrOCRProcessor.from_pretrained("moonstripe/chickenscratch-ocr")
5model = VisionEncoderDecoderModel.from_pretrained("moonstripe/chickenscratch-ocr")
6
7# Load a handwritten line image
8image = Image.open("line.png").convert("RGB")
9
10# Process and predict
11pixel_values = processor(images=image, return_tensors="pt").pixel_values
12generated_ids = model.generate(pixel_values)
13text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
14print(text)