Update app.py

app.py

@@ -2,12 +2,12 @@ import gradio as gr
 import joblib
 
 # Load your saved model
-# model = joblib.load("ann_model.joblib")
+model = joblib.load("ann_model.joblib")
 
 # # Define the prediction function
-def predict(age, hours, education, capital_gain, capital_loss):
-#     features = [[age, hours, education, capital_gain, capital_loss]]
-#     prediction = model.predict(features)
+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]]
+    prediction = model.predict(features)
     prediction = 1
     return "Income >50K" if prediction == 1 else "Income <=50K"