Update app.py

app.py

@@ -26,19 +26,14 @@ unique_values = joblib.load('unique_values.joblib')
     
 unique_class =  unique_values["workclass"]
 unique_education =  unique_values["education"]
-#unique_marital_status =  unique_values["marital_status"]
+unique_marital_status =  unique_values["marital.status"]
 unique_relationship =  unique_values["relationship"]
 unique_occupation =  unique_values["occupation"]
-#unique_sex =  unique_values["sex"]
+unique_sex =  unique_values["sex"]
 unique_race = unique_values["race"]
 unique_country =  unique_values["native.country"]
 print(list(unique_values.keys()))
 
-#if "marital.status" in unique_values:
-    #unique_marital_status = unique_values["marital.status"]
-#else:
-    #print("Key 'marital.status' not found in unique_values")
-
 def main():
     st.title("Adult Income")
 
@@ -48,11 +43,11 @@ def main():
         fnlwgt = st.selectbox("fnlwgt", options=unique_fnlwgt)  # ต้องกำหนด unique_fnlwgt ก่อนใช้
         education = st.selectbox("Education", options=unique_education)
         educational_num = st.selectbox("Educational Num", options=unique_education_num)  # ต้องกำหนด unique_education_num ก่อนใช้
-        #marital_status = st.selectbox("Marital Status", options=unique_marital_status)
+        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)
-        gender = st.selectbox("Gender", options=unique_gender)  # ต้องกำหนด unique_gender ก่อนใช้
+        sex = st.selectbox("SEX", options=unique_sex)  # ต้องกำหนด unique_gender ก่อนใช้
         capital_gain = st.selectbox("Capital Gain", options=unique_capital_gain)  # ต้องกำหนด unique_capital_gain ก่อนใช้
         capital_loss = st.selectbox("Capital Loss", options=unique_capital_loss)  # ต้องกำหนด unique_capital_loss ก่อนใช้
         hours_per_week = st.slider("Hours per week", min_value=1, max_value=100)
@@ -66,11 +61,11 @@ if clicked:
                                          "fnlwgt": [fnlwgt],
                                          "education": [education_encoded],
                                          "educational-num": [educational_num],
-                                         #"marital.status": [marital_status],
+                                         "marital.status": [marital_status],
                                          "occupation": [occupation],
                                          "relationship": [relationship],
                                          "race": [race],
-                                         "gender": [gender],
+                                         "sex": [sex],
                                          "capital_gain": [capital_gain],
                                          "capital_loss": [capital_loss],
                                          "hours.per.week": [hours_per_week],