Views
No views yet
TrOCR on a custom dataset for handwritten text extraction from scanned documents.pytorch_model.bin: Model weights (2.1 GB)config.json, tokenizer_config.json, etc.1from transformers import VisionEncoderDecoderModel, TrOCRProcessor
2from PIL import Image
3import torch
4
5# Load processor and model
6processor = TrOCRProcessor.from_pretrained("Gitesh2003/MESA_TrOCR")
7model = VisionEncoderDecoderModel.from_pretrained("Gitesh2003/MESA_TrOCR")
8
9# Load image
10image = Image.open("sample_image.jpg").convert("RGB")
11
12# OCR
13pixel_values = processor(images=image, return_tensors="pt").pixel_values
14generated_ids = model.generate(pixel_values)
15generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
16
17print(generated_text)