diabeties / app.py
danielle2003's picture
read
519f2ba
raw
history blame
1.14 kB
import streamlit as st
import joblib
import pandas as pd
# Load trained model
model = joblib.load("diabetes-ml/diabetes_model.joblib")
# Streamlit UI
st.title("Diabetes Prediction App")
st.write("Enter patient data to predict diabetes")
# Input fields
st.header("welcome to the app")
pregnancies = st.number_input("Pregnancies", min_value=0, max_value=20)
glucose = st.number_input("Glucose Level", min_value=0, max_value=200)
blood_pressure = st.number_input("Blood Pressure", min_value=0, max_value=200)
skin_thickness = st.number_input("Skin Thickness", min_value=0, max_value=100)
insulin = st.number_input("Insulin Level", min_value=0, max_value=900)
bmi = st.number_input("BMI", min_value=0.0, max_value=60.0)
dpf = st.number_input("Diabetes Pedigree Function", min_value=0.0, max_value=3.0)
age = st.number_input("Age", min_value=0, max_value=120)
# Prediction
if st.button("Predict"):
features = [[pregnancies, glucose, blood_pressure, skin_thickness, insulin, bmi, dpf, age]]
prediction = model.predict(features)[0]
result = "Diabetic" if prediction == 1 else "Not Diabetic"
st.write(f"Prediction: **{result}**")