Spaces:
Runtime error
Runtime error
File size: 2,900 Bytes
b118db8 4fa2a27 b118db8 ffe0bde b118db8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 | 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_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")
with st.form("questionaire"):
gender = st.selectbox("Sex", options=unique_gender)
age = st.slider("Age", min_value=5, max_value=100)
hypertension = st.radio("Hypertension", ('Have Hypertension',"Doesn't Have Hypertension"))
if hypertension == 'Have Hypertension':
st.write('You selected have Hypertension.')
hypertension = 1
else:
st.write("You selected doesn't have Hypertension.")
hypertension = 0
heart_disease = st.radio("Heart Disease", ('Have Heart Disease', "Doesn't Have Heart Disease",))
if heart_disease == 'Have Heart Disease':
st.write('You selected have Heart Disease.')
heart_disease = 1
else:
st.write("You selected doesn't have Heart Disease.")
heart_disease = 0
ever_married = st.selectbox("Married", options=unique_ever_married)
work_type = st.selectbox("Work Type", options=unique_work_type )
Residence_type = st.selectbox("Residence Type", options=unique_Residence_type )
avg_glucose_level = st.number_input("Avg Glucose Level", min_value=20.0, max_value=300.0)
bmi = st.number_input('BMI:', min_value=10.0, max_value=50.0)
smoking_status = st.selectbox("Smoking Status", options=unique_smoking_status )
# clicked==True only when the button is clicked
clicked = st.form_submit_button("Predict Stroke")
if clicked:
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 = 'You had a Stroke' if result[0] == 1 else "You didn't have a Stroke"
st.success("Your predicted is "+result)
if __name__ == "__main__":
main() |