ocr-model / README.md
jasong03's picture
Upload README.md with huggingface_hub
d6b08e4 verified
# 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
```python
import torch
# Load using original filename
model = torch.load("final_model.pt")
# Or using HuggingFace convention
model = torch.load("pytorch_model.bin")
model.eval()
```