File size: 337 Bytes
29467f3 | 1 2 3 4 5 6 7 8 9 10 11 | """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))
|