Update app.py

app.py

@@ -30,18 +30,35 @@ def app_design():
 
     EmploymentType = st.text_input("Employment Type")
     EmploymentType = transform([EmploymentType])
-    
-    GraduateOrNot = st.number_input("Graduate Or Not")
+    sex = st.selectbox("Sex",('Male','Female'))
+    if sex == 'Male':
+        sex = 1
+    else:
+        sex = 0
+    GraduateOrNot = st.selectbox("Graduate Or Not",('Yes','No'))
+    if GraduateOrNot == 'Yes':
+        GraduateOrNot = 1
+    else:
+        GraduateOrNot = 0
 
     AnnualIncome = st.number_input("Annual Income")
 
 
     FamilyMembers = st.number_input("Family Members")
     
-    FrequentFlyer = st.number_input("Frequent Flyer")
-    
+    FrequentFlyer = st.selectbox("Frequent Flyer",('Yes','No'))
+    if FrequentFlyer == 'Yes':
+        FrequentFlyer = 1
+    else:
+        FrequentFlyer = 0
+        
     EverTravelledAbroad = st.number_input("Ever Travelled Abroad")
-    
+    EverTravelledAbroad = st.selectbox("Ever Travelled Abroad",('Yes','No'))
+    if EverTravelledAbroad == 'Yes':
+        EverTravelledAbroad = 1
+    else:
+        EverTravelledAbroad = 0
+
     # Create a feature list from the user inputs
     features = [[Age, EmploymentType, GraduateOrNot, AnnualIncome, FamilyMembers, FrequentFlyer, EverTravelledAbroad]]