Team Demo Resnet

Required loading interface

from transformers import AutoImageProcessor, AutoModelForImageClassification

processor = AutoImageProcessor.from_pretrained(repo_id, revision=commit_hash)
model = AutoModelForImageClassification.from_pretrained(
    repo_id,
    revision=commit_hash,
    trust_remote_code=False,
)

--- library_name: transformers pipeline_tag: image-classification tags: - image-classification - resnet - hackathon --- # ResNet-50 ImageNet-20 Submission Demo This is a Hugging Face-compatible image-classification submission. ## Labels apple, orange, bee, butterfly, camel, elephant, lion, tiger, squirrel, shark, turtle, crab, lobster, bicycle, pickup_truck, tractor, train, keyboard, television, wardrobe ## Original ImageNet class indices used for filtering {'apple': 948, 'orange': 950, 'bee': 309, 'butterfly': 323, 'camel': 354, 'elephant': 386, 'lion': 291, 'tiger': 292, 'squirrel': 335, 'shark': 2, 'turtle': 33, 'crab': 118, 'lobster': 122, 'bicycle': 444, 'pickup_truck': 717, 'tractor': 866, 'train': 820, 'keyboard': 508, 'television': 851, 'wardrobe': 893} ## Required loading interface ```python from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained(repo_id, revision=commit_hash) model = AutoModelForImageClassification.from_pretrained( repo_id, revision=commit_hash, trust_remote_code=False, ) ```