matsammut commited on
Commit
3ac8225
·
verified ·
1 Parent(s): b18aa7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- 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
 
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