Upload 2 files

heart_model.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:203374e75f1e02423a55403a251778170d2e48072ca2a0b7f38481231349358e
+size 1338

model.py

@@ -0,0 +1,65 @@
+import streamlit as st
+import pandas as pd
+import joblib
+
+
+st.title("Heart Disease Prediction")
+
+gender = st.radio("Select Your Gender",["Male","Female"])
+
+if gender == "Male":
+    gender = 1
+else:
+    gender = 0
+
+age = st.slider("Select Your Age")
+
+cigsPerDay = st.number_input("Enter No of Ciggarettes Per Day")
+
+BPMeds = st.radio("Do you take BP Medicines",["True","False"])
+
+if BPMeds == "True":
+    BPMeds = 1.0
+else:
+    BPMeds = 0.0
+
+prevStroke = st.radio("Do you have any prevalent stroke",["True","False"])
+
+if prevStroke=="True":
+    prevStroke=1
+else:
+    prevStroke=0
+
+prevalentHyp = st.radio("Do you have any prevalent Hypertension",["True","False"])
+
+if prevalentHyp=="True":
+    prevalentHyp=1
+else:
+    prevalentHyp=0
+
+diabetes = st.radio("Do you have diabetes",["True","False"])
+
+if diabetes=="True":
+    diabetes=1
+else:
+    diabetes=0
+
+totchol = st.number_input("Enter your Cholestrol Level")
+
+sysBP = st.number_input("Enter your systolic Blood Pressure Level")
+
+diaBP = st.number_input("Enter your Diastolic Blood Pressure Level")
+
+bmi = st.number_input("Enter your BMI")
+
+heart_rate = st.number_input("Enter your Heart Beats per minute")
+
+if st.button("Predict"):
+    model = joblib.load("heart_model.h5")
+    prediction = model.predict([[gender,age,cigsPerDay,BPMeds,prevStroke,prevalentHyp,diabetes,totchol,sysBP,diaBP,bmi,heart_rate]])
+    if prediction[0] ==0:
+        prediction = "You are Not At Risk of getting Heart Disease"
+        st.success(prediction)
+    else:
+        prediction = "You are at Risk of getting Heart Disease"
+        st.success(prediction)