Update app.py

app.py

@@ -11,6 +11,7 @@ model = joblib.load("ann_model.joblib")
 # # Define the prediction function
 def predict(age, workclass, education, marital_status, occupation, relationship, race, gender, capital_gain, capital_loss, hours_per_week, native_country):
     features = [age, workclass, education, marital_status, occupation, relationship, race, gender, capital_gain, capital_loss, hours_per_week, native_country]
+    fixed_features = cleaning_features(features)
     prediction = model.predict(features)
     prediction = 1
     return "Income >50K" if prediction == 1 else "Income <=50K"