Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ def predict(age, workclass, education, marital_status, occupation, relationship,
|
|
18 |
"hours-per-week":[hours_per_week], "native-country":[native_country]}
|
19 |
df = pd.DataFrame(data=columns)
|
20 |
fixed_features = cleaning_features(df)
|
21 |
-
|
22 |
ann_model = pickle.load(ann_model_file)
|
23 |
prediction = ann_model.predict(fixed_features)
|
24 |
# prediction = 1
|
|
|
18 |
"hours-per-week":[hours_per_week], "native-country":[native_country]}
|
19 |
df = pd.DataFrame(data=columns)
|
20 |
fixed_features = cleaning_features(df)
|
21 |
+
with open('ann_model.pkl', 'rb') as ann_model_file:
|
22 |
ann_model = pickle.load(ann_model_file)
|
23 |
prediction = ann_model.predict(fixed_features)
|
24 |
# prediction = 1
|