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