compro-fastapi / model /ocr_loader.py
faizmaula's picture
Upload 72 files
0aec844 verified
raw
history blame contribute delete
501 Bytes
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())