Update app.py

app.py

@@ -1,65 +1,46 @@
 import joblib
 import pandas as pd
+import streamlit as st
 
+model = joblib.load("model.joblib")#
+unique_values = joblib.load("unique_values.joblib")#    
 
-EDU_DICT = {'Preschool': 1,
-            '1st-4th': 2,
-            '5th-6th': 3,
-            '7th-8th': 4,
-            '9th': 5,
-            '10th': 6,
-            '11th': 7,
-            '12th': 8,
-            'HS-grad': 9, 
-            'Some-college': 10,
-            'Assoc-voc': 11,
-            'Assoc-acdm': 12,
-            'Bachelors': 13,
-            'Masters': 14,
-            'Prof-school': 15,
-            'Doctorate': 16
-            }
-
-model = #
-unique_values = #
-    
-unique_class =  unique_values["workclass"]
-unique_education =  unique_values["education"]
-unique_marital_status =  unique_values["marital.status"]
-unique_relationship =  unique_values["relationship"]
-unique_occupation =  unique_values["occupation"]
-unique_sex =  unique_values["sex"]
-unique_race = unique_values["race"]
-unique_country =  unique_values["native.country"]
+unique_gender =  unique_values["gender"]
+unique_hypertension =  unique_values["hypertension"]
+unique_heart_disease =  unique_values["heart_disease"]
+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")
-
+    st.title("Stroke Prediction")
+    
     with st.form("questionaire"):
-        age = # user's input
-        workclass = # user's input
-        education = # user's input
-        Marital_Status = # user's input
-        occupation = # user's input
-        relationship = # user's input
-        race = # user's input
-        sex = # user's input
-        hours_per_week = # user's input
-        native_country = # user's input
+        Gender = st.selectbox("Gender", options=unique_gender)
+        Age = st.slider("Age", min_value=10, max_value=100)
+        Hypertension = st.selectbox("Hypertension", options=unique_hypertension)
+        Heart_Disease = st.selectbox("Heart_Disease", options=unique_heart_disease)
+        Work_type = st.selectbox("Work", options=unique_work_type)
+        Residence = st.selectbox("Residence", options=unique_residence_type)
+        Glucose_Level = st.number_input("Glucose_Level")
+        BMI = st.number_input("BMI")
+        Smoking = st.selectbox("Smoking", options=unique_smoking_status)
 
         # clicked==True only when the button is clicked
-        clicked = st.form_submit_button("Predict income")
+        clicked = st.form_submit_button("Predict Stroke")
         if clicked:
-            result=model.predict(pd.DataFrame({"age": [age],
-                                               "workclass": [workclass],
-                                               "education": [EDU_DICT[education]],
-                                               "marital.status": [Marital_Status],
-                                               "occupation": [occupation],
-                                               "relationship": [relationship],
-                                               "race": [race],
-                                               "sex": [sex],
-                                               "hours.per.week": [hours_per_week],
-                                               "native.country": [native_country]}))
+            result=model.predict(pd.DataFrame({"Gender": [Gender],
+                                               "Age": [Age],
+                                               "Hypertension": [Hypertension],
+                                               "Heart_Disease": [Heart_Disease],
+                                               "Work_type": [Work_type],
+                                               "Residence": [Residence],
+                                               "Avg_Glucose_Level": [Glucose_Level],
+                                               "BMI": [BMI],
+                                               "Smoking": [Smoking]}))
             # Show prediction
-
+            result = 'Yes' if result[0] == 1 else 'No'
+            st.success("Your predicted Stroke is ",result)
 # Run main()
+if __name__ == "__main__":
+    main()