Update app.py

app.py

@@ -17,7 +17,11 @@ sft = ['age', 'serum_creatinine','ejection_fraction']
 
 #data_features = datasets.Features({'text': Value('string'), 'label': ClassLabel(names=sft)})
 
-data = datasets.load_dataset("Boadiwaa/heart_data", ignore_verifications = True)
+
+
+#`my_dataset_dictionary['train'][['age', 'serum_creatinine', 'ejection_fraction']]`. Available splits: ['train']"
+
+data = datasets.load_dataset("Boadiwaa/heart_data", {"train": [sft]}, ignore_verifications = True)
 
 Features = sft