med-doc-classifier / example.py
ds-EkaCare's picture
eka-doc-classifier: base1024 hier (flat L2 default + scope/l1 select, quality 1-100 score)
29467f3 verified
Raw
History Blame Contribute Delete
337 Bytes
"""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))