Update prediction.py

prediction.py

@@ -30,7 +30,7 @@ def run():
       binned_creat_serum = st.selectbox('High serum creatinine', (1, 2), index=1)
       binned_cpk = st.selectbox('CPK Value', (1,2,3), index=1)
       binned_ejection = st.selectbox(' Ejection fraction', (1,2,3), index=1)
-      binned_platelets = st.selectbox('Platelets count', (0, 1, 2), index=1)
+      binned_platelets = st.selectbox('Platelets count', (0, 1, 2), index=1, help='0 = low, \n 1 = normal, \n 2 = high ')
       st.markdown('---')
 
       submitted = st.form_submit_button('Predict')
@@ -63,7 +63,7 @@ def run():
 
       # Predict using gbc
       y_pred_inf = gbc_model.predict(data_inf_df)
-      st.write('# Rating: ', str(int(y_pred_inf)))
+      st.write('# Death possibility: ', str(int(y_pred_inf)))
       
 if __name__ == '__main__':
     run()