Update app.py

app.py

@@ -12,48 +12,46 @@ stroke_dict = {0:"No",
                      1: "Yes"}
 
 model = joblib.load('model.joblib')
-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_relationship =  unique_values["relationship"]
-unique_occupation =  unique_values["occupation"]
-unique_sex =  unique_values["sex"]
-unique_race = unique_values["race"]
-unique_country =  unique_values["native.country"]
+unique_values = joblib.load('unique_values.joblib')    
+unique_gender =  unique_values["gender"]
+unique_hypertension =  unique_values["hypertension"]
+unique_heart_disease =  unique_values["heart_disease"]
+unique_ever_married =  unique_values["ever_married"]
+unique_work_type =  unique_values["work_type"]
+unique_Residence_type = unique_values["Residence_type"]
+unique_smoking_status = unique_values["smoking_status"]
 
 def main():
-    st.title("Adult Income")
+    st.title("Stroke Prediction")
 
     with st.form("questionaire"):
+        gender = st.selectbox("Gender" , options=unique_gender)
         age = st.slider("Age" , min_value=10 , max_value=100)
-        workclass = st.selectbox("Workclass" , 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)
+        hypertension= st.selectbox("Do you have Hypertension ? " , options=unique_hypertension)
+        heart_disease= st.selectbox("Do you have any Heart Desease ? " , options=unique_heart_disease)
+        ever_married = st.selectbox("Have you ever married ? " , options=unique_ever_married)
+        work_type= st.selectbox("What kind of job do you do ? " , options=unique_work_type)
+        Residence_type= st.selectbox("Which kind of place do you live in ? " , options=unique_Residence_type)
+        smoking_status= st.selectbox("Do you smoke ? " , options=unique_smoking_status )
+        bmi  = st.number_input(label="What is your BMI" ,step=1.,format="%.2f")
+        avg_glucose_level= st.number_input(label="What is your average glucose level" ,step=1.,format="%.2f")
 
         # clicked==True only when the button is clicked
         clicked = st.form_submit_button("Predict income")
         if clicked:
-            result=model.predict(pd.DataFrame({"age": [age],
-                                               "workclass": [workclass],
-                                               "education": [EDU_DICT[education]],
-                                               "marital.status": [Marital_Status],
-                                               "occupation": [occupation],
-                                               "relationship": [relationship],
-                                               "race": [race],
-                                               "sex": [sex],
-                                               "hours.per.week": [hours_per_week],
-                                               "native.country": [native_country]}))
+            result=model.predict(pd.DataFrame({"gender": [gender],
+                                               "age": [age],
+                                               "hypertension": [hypertension],
+                                               "heart_disease": [heart_disease],
+                                               "ever_married": [ever_married],
+                                               "work_type": [work_type],
+                                               "Residence_type": [Residence_type],
+                                               "avg_glucose_level":[avg_glucose_level],
+                                               "bmi""[bmi],
+                                               "smoking_status":[smoking_status]}))
             # Show prediction
-            result = '>50k' if result[0] == 1 else '<=50k'
-            st.success("Your predicted income is "+result)
+            result = 'You may not have a stroke' if result[0] == 1 else 'You may have a stroke'
+            st.success("Your prediction "+result)
             
 # Run main()
 if __name__ == "__main__" :