| 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' |