Update app.py

app.py

@@ -22,9 +22,15 @@ from PIL import Image
 dataset_name = "saifhmb/CreditCardRisk"
 dataset = load_dataset(dataset_name, split = 'train')
 dataset = pd.DataFrame(dataset)
+dataset['GENDER'] = dataset['GENDER'].str.replace(' ', '')
+dataset['GENDER'] = dataset['GENDER'].replace(['n', 'y'], [0, 1], inplace = True)
 dataset['MARITAL'] = dataset['MARITAL'].str.replace(' ', '')
 dataset['MARITAL'] = dataset['MARITAL'].replace(['married', 'single', 'divsepwid'], [0, 1, 2], inplace = True)
+dataset['HOWPAID'] = dataset['HOWPAID'].str.replace(' ', '')
 dataset['HOWPAID'] = dataset['HOWPAID'].replace(['n', 'y'], [0, 1], inplace = True)
+dataset['MORTGAGE'] = dataset['MORTGAGE'].replace(' ', '')
+dataset['MORTGAGE'] = dataset['MORTGAGE'].replace(['weekly', 'monthly'], [0, 1], inplace = True)
+dataset = dataset.drop(['ID'], axis = 1)
 X = dataset.iloc[:, :-1].values
 y = dataset.iloc[:, -1].values