Update app.py

app.py

@@ -13,11 +13,15 @@ from imblearn.over_sampling import SMOTE
 import gradio as gr
 from datasets import load_dataset
 
-dataset = load_dataset("Boadiwaa/heart_data")
+sft = ['age', 'serum_creatinine','ejection_fraction']
+
+data_features = Features({'text': Value('string'), 'label': ClassLabel(names=sft)})
+
+dataset = load_dataset("Boadiwaa/heart_data", column_names=['text', 'label'], features=sft)
 
 data = dataset
 
-Features = list(('age', 'serum_creatinine','ejection_fraction'))
+Features = sft
 
 x = data[Features]
 y = data["DEATH_EVENT"]