File size: 501 Bytes
0aec844
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import torch
from transformers import TrOCRProcessor, VisionEncoderDecoderModel
import easyocr

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

# === Load ONCE ===
processor = TrOCRProcessor.from_pretrained(
    "microsoft/trocr-base-printed",
    use_fast=False
)

model = VisionEncoderDecoderModel.from_pretrained(
    "microsoft/trocr-base-printed"
).to(device)

# EasyOCR reader (load sekali)
reader = easyocr.Reader(['en'], gpu=torch.cuda.is_available())