Views
No views yet
TrOCRProcessor from Hugging Face to process and generate text from images.syubraj/TrOCR_Nepali1from transformers import VisionEncoderDecoderModel, TrOCRProcessor, AutoTokenizer
2from PIL import Image
3import torch
4
5# Load the model and processor
6tokenizer = AutoTokenizer.from_pretrained("syubraj/TrOCR_Nepali")
7model = VisionEncoderDecoderModel.from_pretrained("syubraj/TrOCR_Nepali")
8processor = TrOCRProcessor.from_pretrained("syubraj/TrOCR_Nepali")
9
10# Load image
11image = Image.open("path_to_image").convert("RGB")
12
13# Preprocess image
14pixel_values = processor(image, return_tensors="pt").pixel_values
15
16# Generate text
17generated_ids = model.generate(pixel_values)
18generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
19
20print(generated_text)