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'