Update prediction.py

prediction.py

@@ -1,16 +1,15 @@
 import streamlit as st
 import pandas as pd
 import joblib
-from sklearn.compose import ColumnTransformer
-from sklearn.pipeline import Pipeline
-from xgboost import XGBClassifier
+
 
 def app():
     st.title("Transaction Data Input")
     st.write("Choose to upload a CSV file or manually input transaction data.")
     
     # Load pre-trained model
-    model = joblib.load('model.pkl')
+    with open('model.pkl', 'rb') as file_1:
+        model = joblib.load(file_1)
 
     # Option to choose upload or manual input
     option = st.radio("Select input method:", ("Upload CSV", "Manual Input"))
@@ -31,7 +30,7 @@ def app():
                 data['prediction'] = data['prediction'].map({1: 'Fraud Transactions', 0: 'Not Fraud Transactions'})
 
                 st.write("Predictions:")
-                st.write(data[['nameOrig', 'nameDest', 'prediction']])
+                st.write(data[['type','nameOrig', 'nameDest', 'prediction']])
             
     elif option == "Manual Input":
         st.write("Manually input data:")
@@ -71,7 +70,7 @@ def app():
             manual_data['prediction'] = manual_data['prediction'].map({1: 'Fraud Transactions', 0: 'Not Fraud Transactions'})
 
             st.write("Predictions:")
-            st.write(manual_data[['nameOrig', 'nameDest', 'prediction']])
+            st.write(manual_data[['type','nameOrig', 'nameDest', 'prediction']])
 
 if __name__ == "__main__":
     app()