9c855c1 9506f43
1
2
3
4
5
6
7
from datasets import load_dataset dataset_name = "jonathan-roberts1/Satellite-Images-of-Hurricane-Damage" ds = load_dataset(dataset_name) labels = ds['train'].features['label'].names UNTRAINED = 'google/vit-base-patch16-224-in21k'