example.py

"""Minimal usage example."""

import torch
from PIL import Image
from huggingface_hub import hf_hub_download
from modeling import load, preprocess, ctc_greedy

path = hf_hub_download("LocalDoc/azerbaijani-ocr-crnn", "model.pt")
model, itos = load(path)

img = Image.open("line.png")          # crop of ONE text line
with torch.no_grad():
    logits = model(preprocess(img))
print(ctc_greedy(logits, itos))