Add app.py for income prediction

app.py

@@ -0,0 +1,36 @@
+import gradio as gr
+import pickle
+import pandas as pd
+
+# Load the model
+with open("salar_xgb_team.pkl", "rb") as f:
+    model = pickle.load(f)
+
+# Define prediction function
+def predict_salary(age, education_num, sex, capital_gain, capital_loss, hours_per_week):
+    # sex: "Male" = 0, "Female" = 1
+    sex_num = 0 if sex == "Male" else 1
+
+    df = pd.DataFrame([[age, education_num, sex_num, capital_gain, capital_loss, hours_per_week]],
+                      columns=['age', 'education-num', 'sex', 'capital-gain', 'capital-loss', 'hours-per-week'])
+
+    pred = model.predict(df)[0]
+    return ">50K" if pred == 1 else "<=50K"
+
+# Create Gradio UI
+demo = gr.Interface(
+    fn=predict_salary,
+    inputs=[
+        gr.Number(label="Age"),
+        gr.Number(label="Education Level (numeric)"),
+        gr.Radio(["Male", "Female"], label="Sex"),
+        gr.Number(label="Capital Gain"),
+        gr.Number(label="Capital Loss"),
+        gr.Number(label="Hours per Week")
+    ],
+    outputs="text",
+    title="Income Prediction App",
+    description="Enter demographic and income-related data to predict if salary is >50K or <=50K"
+)
+
+demo.launch()