Views
No views yet
1from transformers import AutoModelForImageClassification, AutoImageProcessor
2
3# Load ImageNet pretrained (1000 classes)
4model = AutoModelForImageClassification.from_pretrained(
5 "henriquequeirozcunha/microvit-s3",
6 trust_remote_code=True,
7)
8
9# Fine-tune for binary classification
10model = AutoModelForImageClassification.from_pretrained(
11 "henriquequeirozcunha/microvit-s3",
12 num_labels=2,
13 ignore_mismatched_sizes=True,
14 trust_remote_code=True,
15)