Update prediction.py

prediction.py

@@ -84,11 +84,23 @@ def run():
     #Logic ketika predic button ditekan
 
     if submitted:
-        
-        #predict using pipe rf model
-        predictions = loaded_model.predict(data_inf)
-        
-        st.write('## Deposit : ', str(int(predictions)))
+    
+    # Predict using loaded model
+    predictions = loaded_model.predict(data_inf)
+    
+    # Menampilkan data input
+    st.write('## Data Input:')
+    st.dataframe(data_inf)
+
+    # Menampilkan hasil prediksi
+    st.write('## Hasil Prediksi:')
+    st.write('Deposit:', int(predictions[0]))
+
+    # Menampilkan pesan berdasarkan hasil prediksi
+    if predictions[0] == 0:
+        st.write("Nasabah belum berlangganan deposito berjangka.")
+    else:
+        st.write("Nasabah telah berlangganan deposito berjangka.")
 
 if __name__ == '__main__':
    run()