Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -21,6 +21,8 @@ def predict(age, workclass, education, marital_status, occupation, relationship,
|
|
21 |
fixed_features = cleaning_features(df)
|
22 |
# with open('ann_model.pkl', 'rb') as ann_model_file:
|
23 |
# ann_model = pickle.load(ann_model_file)
|
|
|
|
|
24 |
ann_model = load_model('ann_model.h5')
|
25 |
prediction = ann_model.predict(fixed_features)
|
26 |
# prediction = 1
|
|
|
21 |
fixed_features = cleaning_features(df)
|
22 |
# with open('ann_model.pkl', 'rb') as ann_model_file:
|
23 |
# ann_model = pickle.load(ann_model_file)
|
24 |
+
scaler = StandardScaler()
|
25 |
+
X = scaler.fit_transform(fixed_features)
|
26 |
ann_model = load_model('ann_model.h5')
|
27 |
prediction = ann_model.predict(fixed_features)
|
28 |
# prediction = 1
|