Update app.py

app.py

@@ -4,59 +4,52 @@ import pickle
 import streamlit.components.v1 as components
 from sklearn.preprocessing import LabelEncoder
 
-# Load the pickled model
+@st.cache_resource
 def load_model():
-    return pickle.load(open('online_payment_fraud_detection_randomforest.pkl', 'rb'))
+    with open('online_payment_fraud_detection_randomforest.pkl', 'rb') as f:
+        return pickle.load(f)
 
-# Load the LabelEncoder
+@st.cache_resource
 def load_label_encoder():
     with open('label_encoder.pkl', 'rb') as f:
         return pickle.load(f)
 
-# Function for model prediction
 def model_prediction(model, features):
-    predicted = str(model.predict(features)[0])
-    return predicted
+    features = np.array(features)
+    return str(model.predict(features)[0])
 
 def transform(le, text):
-    text = le.transform(text)
-    return text[0]
+    return le.transform(text)[0]
 
 def app_design(le):
-    # Add input fields for High, Open, and Low values
-    # image = 'Ramdevs2'
-    # st.image(image, use_column_width=True)
-    
     st.subheader("Enter the following values:")
 
-    step = st.number_input("Step: represents a unit of time where 1 step equals 1 hour")
+    step = st.number_input("Step: represents a unit of time where 1 step equals 1 hour", value=0)
     typeup = st.selectbox('Type of online transaction', ('PAYMENT', 'TRANSFER', 'CASH_OUT', 'DEBIT', 'CASH_IN'))
     typeup = transform(le, [typeup])
-    amount = st.number_input("The amount of the transaction")
+    amount = st.number_input("The amount of the transaction", value=0.0)
+    
     nameOrig = st.text_input("Transaction ID")
-    nameOrig = transform(le, [nameOrig])
-    oldbalanceOrg = st.number_input("Balance before the transaction")
-    newbalanceOrig = st.number_input("Balance after the transaction")
+    oldbalanceOrg = st.number_input("Balance before the transaction", value=0.0)
+    newbalanceOrig = st.number_input("Balance after the transaction", value=0.0)
+    
     nameDest = st.text_input("Recipient ID")
-    nameDest = transform(le, [nameDest])
-    oldbalanceDest = st.number_input("Initial balance of recipient before the transaction")
-    newbalanceDest = st.number_input("The new balance of recipient after the transaction")
+    oldbalanceDest = st.number_input("Initial balance of recipient before the transaction", value=0.0)
+    newbalanceDest = st.number_input("The new balance of recipient after the transaction", value=0.0)
+    
     isFlaggedFraud = st.selectbox('IsFlaggedFraud', ('Yes', 'No'))
     isFlaggedFraud = transform(le, [isFlaggedFraud])
+
+    features = [[step, typeup, amount, 0, oldbalanceOrg, newbalanceOrig, 0, oldbalanceDest, newbalanceDest, isFlaggedFraud]]
     
-    # Create a feature list from the user inputs
-    features = [[step, typeup, amount, nameOrig, oldbalanceOrg, newbalanceOrig, nameDest, oldbalanceDest, newbalanceDest, isFlaggedFraud]]
-    
-    # Load the model
     model = load_model()
     
-    # Make a prediction when the user clicks the "Predict" button
     if st.button('Predict Online Payment Fraud'):
         predicted_value = model_prediction(model, features)
         if predicted_value == '1':
-            st.success("Online payment fraud detected")
+            st.success("🚨 Online payment fraud detected!")
         else:
-            st.success("No online payment fraud detected")
+            st.success("✅ No online payment fraud detected!")
 
 def about_RamDevs():
     components.html("""