Update app.py

app.py

@@ -11,44 +11,21 @@ import joblib
 import pandas as pd
 import streamlit as st
 model = joblib.load('model.joblib')
-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_residence_type = unique_values["Residence_type"]
-unique_smoking_status = unique_values["smoking_status"]
-unique_work_type = unique_values["work_type"]
 
 def main():
     st.title("Brain stroke predict")
     with st.form("questionaire"):
         age = st.slider("Age",min_value=10, max_value=100)
-        gender = st.selectbox("Gender",options = unique_gender)
-        hypertension = st.selectbox("Hypertension",options = unique_hypertension )
-        heart_disease = st.selectbox("Heart disease",options = unique_heart_disease)
-        ever_married = st.selectbox("Ever married?",options = unique_ever_married)
-        residence_type = st.selectbox("Residence",options = unique_residence_type)
-        smoking_status = st.selectbox("Smoke",options = unique_smoking_status)
-        work_type = st.selectbox("Work type",options = unique_work_type)
         avg_glucose_level = st.slider("Average glucose level",min_value=40, max_value=300)
         bmi = st.slider("BMI(Body mass index )",min_value=12, max_value=50)
         # clicked==True only when the button is clicked
         clicked = st.form_submit_button("Predict")
         if clicked:
             result=model.predict(pd.DataFrame({"age": [age],
-                                               "gender": [gender],
-                                               "hypertension": [hypertension],
-                                               "heart_disease": [heart_disease],
-                                               "ever_married": [ever_married],
-                                               "Residence_type": [residence_type],
-                                               "smoking_status": [smoking_status],
-                                               "work_type": [work_type],        
                                                "avg_glucose_level": [avg_glucose_level],
                                                "bmi": [bmi]}))
             # Show prediction
-         
             if result[0] == 1:
                 st.success("You have change of getting brain stroke")
             else: