Update app.py

app.py

@@ -5,10 +5,11 @@ import joblib
 # model = joblib.load("ann_model.joblib")
 
 # # Define the prediction function
-# def predict(age, hours, education, capital_gain, capital_loss):
+def predict(age, hours, education, capital_gain, capital_loss):
 #     features = [[age, hours, education, capital_gain, capital_loss]]
 #     prediction = model.predict(features)
-#     return "Income >50K" if prediction == 1 else "Income <=50K"
+    prediction = 1
+    return "Income >50K" if prediction == 1 else "Income <=50K"
 
 # Create the Gradio interface
 interface = gr.Interface(