Update app.py

app.py

@@ -1,7 +1,8 @@
 import streamlit as st
 import pandas as pd
-
-
+import pickle
+# import model
+model = pickle.load(open("model.pkl", "rb"))
 #title
 st.title("Predict Death Event")
 st.write("Created by Sihar Pangaribuan")
@@ -35,7 +36,7 @@ data = pd.DataFrame({'age': [age],
                 'time': [time]})
 
 # model predict
-death = rf_randomCV.predict(data).tolist()[0]
+death = model.predict(data).tolist()[0]
 
 # interpretation
 st.write('Predition Result: ')