Upload app.py

app.py

@@ -0,0 +1,71 @@
+# -*- coding: utf-8 -*-
+"""app.ipynb
+
+Automatically generated by Colaboratory.
+
+Original file is located at
+    https://colab.research.google.com/drive/1wBXKa6BYIgr42zBNgmQCoaWlY-jwMxZg
+"""
+
+import joblib
+import pandas as pd
+import streamlit as st
+
+model = joblib.load('model.joblib')
+unique_values = joblib.load('unique_values.joblib')
+
+unique_gender =  unique_values["gender"]
+unique_ever_married =  unique_values["ever_married"]
+unique_work_type =  unique_values["work_type"]
+unique_Residence_type =  unique_values["Residence_type"]
+unique_smoking_status =  unique_values["smoking_status"]
+
+def main():
+    st.title("Adult Income")
+
+    with st.form("questionaire"):
+        gender = st.selectbox("Sex", options=unique_gender)
+        age = st.slider("Age", min_value=5, max_value=100)
+
+        hypertension = st.radio("Hypertension", ('Have Hypertension',"Doesn't Have Hypertension"))
+        if hypertension == 'Have Hypertension':
+            st.write('You selected have Hypertension.')
+            hypertension = 1
+        else:
+            st.write("You selected doesn't have Hypertension.")
+            hypertenstion = 0
+
+        heart_disease = st.radio("Heart Disease", ('Have Heart Disease', "Doesn't Have Heart Disease",))
+        if heart_disease == 'Have Heart Disease':
+            st.write('You selected have Heart Disease.')
+            heart_disease = 1
+        else:
+            st.write("You selected doesn't have Heart Disease.")
+            heart_disease = 0
+
+        ever_married = st.selectbox("Married", options=unique_ever_married)
+        work_type = st.selectbox("Work Type", options=unique_work_type )
+        Residence_type = st.selectbox("Residence Type", options=unique_Residence_type )
+        avg_glucose_level = st.number_input("Avg Glucose Level", min_value=20.0, max_value=300.0)
+        bmi = st.number_input('BMI:', min_value=10.0, max_value=50.0)
+        smoking_status =  st.selectbox("Smoking Status", options=unique_smoking_status )
+
+        # clicked==True only when the button is clicked
+        clicked = st.form_submit_button("Predict Stroke")
+        if clicked:
+            result=model.predict(pd.DataFrame({"gender": [gender],
+                                               "age": [age],
+                                               "hypertension": [hypertension],
+                                               "heart_disease": [heart_disease],
+                                               "ever_married": [ever_married],
+                                               "work_type": [work_type],
+                                               "Residence_type": [Residence_type],
+                                               "avg_glucose_level": [avg_glucose_level],
+                                               "bmi": [bmi],
+                                               "smoking_status": [smoking_status]}))
+            # Show prediction
+            result = 'You had a Stroke' if result[0] == 1 else "You didn't have a Stroke"
+            st.success("Your predicted income is "+result)
+            
+if __name__ == "__main__":
+    main()