File size: 278 Bytes
9c855c1 9506f43 7a1b8c6 13711f9 |
1 2 3 4 5 6 7 8 9 |
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'
TRAINED = 'till-onethousand/hurricane_model' |