Update prediction.py

prediction.py

@@ -85,7 +85,9 @@ def run():
 
         # Predict
         y_pred_inf = rfc.predict(data_inf_final)
-        st.write('# Prediksi : ', str(int(y_pred_inf)))
-        st.write('Jika prediksi menunjukan angka 1 maka ada kemungkinan booking tersebut akan di cancel oleh pelanggan')
+        prediction = 'Not Cancel' if y_pred_inf == 0 else 'Cancel'
+
+        st.write('# Prediction: ', prediction)
+        st.write('If the prediction shows "Cancel", there is a possibility that the booking will be canceled by the customer.')
 
 if __name__ == '__main__': run()