Spaces:
Sleeping
Sleeping
File size: 996 Bytes
c6cc915 |
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 |
import streamlit as st
import pickle
from PIL import Image
ok = Image.open('ok.jpg')
not_ok = Image.open("notok.jpg")
st.write ('''
## Heart Health Predictor
''')
age = st.number_input('Enter your age', min_value= 0, max_value= 100, step = 1)
gender = st.selectbox(
'Gender',
('Male', 'Female'))
if gender == "Male":
gender_value = 1
else:
gender_value = 0
bp = st.number_input('Enter your BP', value= 120, step = 1)
cholestrol = st.number_input('Enter your Cholestrol', value= 100, step = 1)
data = [[age, gender_value, bp, cholestrol]]
#load models
# @st.cache_data(allow_output_mutation = True)
def cache_model(path):
model = pickle.load(open(path, 'rb'))
return (model)
heart_model = cache_model("model.pkl")
if st.button('Predict'):
result = heart_model.predict(data)
if result == 1:
st.image(not_ok)
st.write("## Please visit a cardiologist")
else:
st.image(ok)
st.write("## Great!, You are at no risk")
|