Update app.py

app.py

@@ -21,7 +21,7 @@ def predict(age, workclass, education, marital_status, occupation, relationship,
     "race":[race], "gender":[gender], "capital-gain":[capital_gain], "capital-loss":[capital_loss], 
     "hours-per-week":[hours_per_week], "native-country":[native_country]}
     df = pd.DataFrame(data=columns)
-    fixed_features = cleaning_features(df)
+    fixed_features = cleaning_features(df,race)
     # with open('ann_model.pkl', 'rb') as ann_model_file:
     #     ann_model = pickle.load(ann_model_file)
     scaler = StandardScaler()
@@ -31,7 +31,7 @@ def predict(age, workclass, education, marital_status, occupation, relationship,
     # prediction = 1
     return "Income >50K" if prediction == 1 else "Income <=50K"
 
-def cleaning_features(data):
+def cleaning_features(data,race):
     # with open('race_onehot_encoder.pkl', 'rb') as enc_file:
     #     encoder = pickle.load(enc_file)