README.md

OCR Model

This is an OCR model trained with PyTorch Lightning.

Files

• final_model.pt: Original model file
• pytorch_model.bin: Same model with HuggingFace naming convention

Usage

import torch

# Load using original filename
model = torch.load("final_model.pt")

# Or using HuggingFace convention
model = torch.load("pytorch_model.bin")

model.eval()