Upload prediction.py

prediction.py

@@ -28,7 +28,7 @@ def run():
 
       
      
-      submitted = st.form_submit_button(' Still Alive or Not ?')
+      submitted = st.form_submit_button('Is the patient still alive?')
 
   df_inf = {
       'age': age,
@@ -58,7 +58,11 @@ def run():
   if submitted:
       # Predict using RandomForest
       y_pred_inf = rf_gridcv_best.predict(df_inf_final)
-      st.write('# Still Alive or Not ? (0 = Alive, 1 = Deceased) ',str(int(y_pred_inf)))
+      st.write('# Is the patient still alive ?')
+      if y_pred_inf == 0:
+         st.subheader('Still Alive (^o^)/ ')
+      else:
+         st.subheader('Died from heart failure (T_T)')
 
 if __name__ == '__main__':
     run()