Update app.py

app.py

@@ -24,52 +24,42 @@ model = joblib.load('model.joblib')
 unique_values = joblib.load('unique_values.joblib')
 
 unique_class =  unique_values["workclass"]
-unique_fnlwgt = unique_values["fnlwgt"]
 unique_education =  unique_values["education"]
-unique_education_num =  unique_values["educational_num"]
 unique_marital_status =  unique_values["marital.status"]
-unique_occupation=  unique_values["occupation"]
 unique_relationship =  unique_values["relationship"]
+unique_occupation =  unique_values["occupation"]
+unique_gender =  unique_values["gender"]
 unique_race = unique_values["race"]
-unique_gender = unique_values["gender"]
-unique_capital_gain = unique_values["capital_gain"]
-unique_capital_loss = unique_values["capital_loss"]
-unique_hours_per_week = unique_values["hours_per_week"]
 unique_country =  unique_values["native.country"]
-unique_income =  unique_values["income"]
 
 def main():
     st.title("Adult Income Analysis")
 
     with st.form("questionaire"):
-        age = st.slider("Age", min_value=10, max_value=100)
-        workclass = st.selectbox("Workclass", unique_class)
-        fnlwgt = st.selectbox("fnlwgt",unique_fnlwgt)
-        education = st.selectbox("Education", unique_education)
-        educational_num = st.selectbox("educational-num",unique_education_num)
-        Marital_Status = st.selectbox("Marital Status", unique_marital_status)
-        occupation = st.selectbox("Occupation", unique_occupation)
-        relationship = st.selectbox("Relationship", unique_relationship)
-        race = st.selectbox("Race", unique_race)
-        gender = st.selectbox("gender", unique_gender)
-        capital_gain = st.selectbox("capital_gain", unique_gender)
-        capital_loss = st.selectbox("capital_loss", unique_gender)
-        hours_per_week = st.slider("Hours per week", min_value=1, max_value=100)
-        native_country = st.selectbox("Country", unique_country)
+        age = st.slider("Age",min_value = 10 , max_value=100)
+        workclass = st.selectbox("Workcalss", options = unique_class)
+        education = st.selectbox("Education", options = unique_education)
+        Marital_Status = st.selectbox("Marital status", options = unique_marital_status)
+        occupation = st.selectbox("Occupation", options = unique_occupation)
+        relationship = st.selectbox("Relationship", options = unique_relationship)
+        race = st.selectbox("Race", options = unique_race)
+        sex =st.selectbox("Sex", options = unique_sex)
+        hours_per_week =st.slider("Hours per week",min_value = 1 , max_value=100)
+        native_country =st.selectbox("Native country", options = unique_country)
 
         clicked = st.form_submit_button("Predict income")
         if clicked:
             result=model.predict(pd.DataFrame({"age": [age],
-                                               "workclass": [workclass],"fnlwgt": [fnlwgt],                                               
+                                               "workclass": [workclass],
                                                "education": [EDU_DICT[education]],
-                                               "educational-num": [educational_num],
                                                "marital.status": [Marital_Status],
                                                "occupation": [occupation],
                                                "relationship": [relationship],
-                                               "race": [race],
-                                               "gender": [gender],"capital_gain": [capital_gain],"capital_loss": [capital-loss],
+                                               "gender": [gender],
+                                               "sex": [sex],
                                                "hours.per.week": [hours_per_week],
                                                "native.country": [native_country]}))
+            
             result = '>50K' if result[0] == 1 else '<=50K'
             st.success('The predicted income is {}'.format(result))