File size: 234 Bytes
9506f43 |
1 2 3 4 5 6 7 |
import 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' |