Update app.py

app.py

@@ -53,7 +53,7 @@ def main():
                                                "smoking": [smoking],
                                                "time": [time]}))
             # Show prediction
-            result = '1' if result[0] == 1 else '0'
+            result = 'DEATH' if result[0] == 1 else 'NO DEATH'
             st.success("Your prediction death is ", result)
 
 # Run main()