"""Minimal usage example. Run: python example.py path/to/image.jpg"""
import sys
from transformers import AutoModel
from PIL import Image

model = AutoModel.from_pretrained(".", trust_remote_code=True).eval()
img = Image.open(sys.argv[1] if len(sys.argv) > 1 else "doc.jpg")

import json
print(json.dumps(model.classify(img), indent=2))