Spaces:
Sleeping
Sleeping
import joblib | |
import pandas as pd | |
import streamlit as st | |
model = joblib.load('model.joblib') | |
unique_values = joblib.load('unique_values.joblib') | |
unique_hypertension = unique_values["hypertension"] | |
unique_heart_disease = unique_values["heart_disease"] | |
unique_bmi = unique_values["bmi"] | |
unique_HbA1c_level = unique_values["HbA1c_level"] | |
unique_blood_glucose_level = unique_values["blood_glucose_level"] | |
unique_gender = unique_values["gender"] | |
unique_smoking_history = unique_values["smoking_history"] | |
def main(): | |
st.title("diabetes prediction") | |
with st.form("questionaire"): | |
age = st.slider("Age", min_value=10, max_value=100) | |
hypertension = st.selectbox("Hypertemsion", unique_hypertension) | |
heart_disease = st.selectbox("Heart_desease", unique_heart_disease) | |
bmi = st.selectbox("Bmi", unique_bmi) | |
HbA1c_level = st.selectbox("HbA1c_level", unique_HbA1c_level) | |
blood_glucose_level = st.selectbox("Blood_glucose_level", blood_glucose_level) | |
gender = st.selectbox("Gender", gender) | |
smoking_history = st.selectbox("Smoking_history", smoking_history) | |
clicked = st.form_submit_button("Predict diabetes") | |
if clicked: | |
result=model.predict(pd.DataFrame({"age": [age], | |
"hypertension": [hypertension], | |
"heart_disease": [heart_disease], | |
"bmi": [bmi], | |
"HbA1c_level": [HbA1c_level], | |
"blood_glucose_level": [blood_glucose_level], | |
"gender": [gender], | |
"smoking_history": [smoking_history]})) | |
result = '0' if result[0] == 1 else '1' | |
st.success('The predicted income is {}'.format(result)) | |
if __name__=='__main__': | |
main() | |