Spaces:
Runtime error
Runtime error
File size: 272 Bytes
c903d33 da26c6d c903d33 da26c6d c903d33 da26c6d c903d33 |
1 2 3 4 5 6 7 8 9 |
from transformers import ViTFeatureExtractor, ViTForImageClassification
def create_ViT():
extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224")
model = ViTForImageClassification.from_pretrained("google/vit-base-patch16-224")
return model
|