GS123/heart_predictor

1import streamlit as st2import pickle3from PIL import Image4 5 6ok = Image.open('ok.jpg')7not_ok = Image.open("notok.jpg")8 9 10st.write ('''11 12## Heart Health Predictor13''')14 15age = st.number_input('Enter your age', min_value= 0, max_value= 100, step = 1)16 17gender = st.selectbox(18    'Gender',19    ('Male', 'Female'))20 21if gender == "Male":22    gender_value = 123else:24    gender_value = 025 26bp = st.number_input('Enter your BP', value= 120, step = 1)27cholestrol = st.number_input('Enter your Cholestrol', value= 100, step = 1)28 29 30data = [[age, gender_value, bp, cholestrol]]31 32#load models33# @st.cache_data(allow_output_mutation = True)34def cache_model(path):35    model = pickle.load(open(path, 'rb'))36    return (model)37 38heart_model = cache_model("model.pkl")39 40if st.button('Predict'):41    result = heart_model.predict(data)42    if result == 1:43        st.image(not_ok)44        st.write("## Please visit a cardiologist")45 46    else:47        st.image(ok)48        st.write("## Great!, You are at no risk")49 50 51 52