Upload app.py

app.py

@@ -102,20 +102,24 @@ def get_input_data():
         col_age, col_sex = st.columns(2)
         with col_age:
             session.input_data.loc[0, 'age'] = st.slider('Age', 18, 75)
-            #on_change=update_prediction(session.input_data)
+            # on_change=update_prediction(session.input_data)
         with col_sex:
             sex = st.radio('Sex', ['Female', 'Male'])
             st.write('<style>div.row-widget.stRadio > div{flex-direction:row;}\
                      </style>', unsafe_allow_html=True)
-            session.input_data.loc[0, 'gender'] = [1 if 'Female' in sex \
+            session.input_data.loc[0, 'gender'] = [1 if 'Female' in sex
                                                    else 2][0]
 
         # Row 2
         col_height, col_weight = st.columns(2)
         with col_height:
             session.input_data.loc[0, 'height'] = st.slider('Height', 140, 200)
+            session.input_data["height"] = session.input_data["height"].astype(
+                int)
         with col_weight:
             session.input_data.loc[0, 'weight'] = st.slider('Weight', 40, 140)
+            session.input_data["weight"] = session.input_data["weight"].astype(
+                int)
 
         # Row 3
         col_ap_hi, col_ap_lo = st.columns(2)
@@ -134,26 +138,37 @@ def get_input_data():
             session.input_data.loc[0, 'cholesterol'] = [
                 1 if 'Normal' in cholest else 2 if 'Above normal' in cholest
                 else 3][0]
+            session.input_data["cholesterol"] = session.input_data["cholesterol"].astype(
+                'category')
+
         with col_gluc:
             gluc = st.radio(
                 'Glucose', ['Normal', 'Above normal', 'Well above normal'])
             session.input_data.loc[0, 'gluc'] = [
                 1 if 'Normal' in gluc else 2 if 'Above normal' in gluc
                 else 3][0]
+            session.input_data["gluc"] = session.input_data["gluc"].astype(
+                'category')
 
         # Row 5
         col_alco, col_smk = st.columns(2)
         with col_alco:
             alco = st.radio('Alcohol intake', ['Yes', 'No'])
             session.input_data.loc[0, 'alco'] = [1 if 'Yes' in alco else 2][0]
+            session.input_data["alco"] = session.input_data["alco"].astype(
+                bool)
         with col_smk:
             smoke = st.radio('Smoking', ['Yes', 'No'])
-            session.input_data.loc[0, 'smoke'] = [1 if 'Yes' in smoke \
+            session.input_data.loc[0, 'smoke'] = [1 if 'Yes' in smoke
                                                   else 2][0]
+            session.input_data["smoke"] = session.input_data["smoke"].astype(
+                bool)
 
         # Row 6
         active = st.radio('Physical activity', ['Yes', 'No'])
         session.input_data.loc[0, 'active'] = [1 if 'Yes' in active else 2][0]
+        session.input_data["active"] = session.input_data["active"].astype(
+            bool)
 
         st.write("")
 
@@ -179,10 +194,10 @@ def generate_prediction(input_data):
     current_time = datetime.now().strftime("%H:%M:%S")
     st.write("Current Time =", current_time)
     st.write(input_data)
-    
+
     prediction = MODEL.predict(input_data)
     st.write(prediction)
-    
+
     return prediction
 
 
@@ -233,7 +248,6 @@ if __name__ == "__main__":
         # current_time = datetime.now().strftime("%H:%M:%S")
         # st.write("Current Time =", current_time)
         # st.write(session.input_data)
-        
 
     st.text("")
     st.text("")