Update app.py

app.py

@@ -14,8 +14,8 @@ from datasets import load_dataset
 dataset = load_dataset("Hemg/brain-tumour-dataset")
 
 # Split the dataset into features (X) and labels (y)
-X = dataset['images']
-y = dataset['labels']
+X = dataset['image']
+y = dataset['label']
 
 # Preprocess labels
 y = to_categorical(y, num_classes=num_classes)