Update app.py

app.py

@@ -39,11 +39,23 @@ data = pd.DataFrame({'age': [age],
 
 data = preproses.transform(data)
 # model predict
-death = model.predict(data).tolist()[0]
+
+if st.button('Predict'):
+    prediction = model.predict(data).tolist()[0]
+
+    if prediction == 1:
+        prediction = 'Death'
+    else:
+        prediction = 'Live'
+
+    st.write('Based on user input, predicted: ')
+
+
+
 
 # interpretation
-st.write('Predition Result: ')
-if death == 0:
-    st.text('live')
-else:
-    st.text('Death')
+# st.write('Predition Result: ')
+# if death == 0:
+#     st.text('live')
+# else:
+#     st.text('Death')