Update app.py

app.py

@@ -5,36 +5,39 @@ import tensorflow as tf
 import joblib
 
 # Load trained model
-model = tf.keras.models.load_model("banking_model.keras")
+model = tf.keras.models.load_model("baking_model.keras")
 
 # Load encoders and scaler
 label_encoders = joblib.load("label_encoders.pkl")
 scaler = joblib.load("scaler.pkl")
 
-# Define the input features
-feature_names = [
-    # Add all the feature column names used in training
-]
+# Define feature names
+numerical_features = ["DPD", "Credit Expiration"]
+binary_features = ["Feature1", "Feature2", "Feature3"]  # Replace with actual binary features
+stage_feature = "Stage As Last Month"
 
 st.title("Classification Prediction App")
 
 # Create input fields for user input
 user_input = {}
-for feature in feature_names:
-    if feature in label_encoders:  # If it's a categorical feature
-        options = list(label_encoders[feature].classes_)
-        user_input[feature] = st.selectbox(f"Select {feature}", options)
-    else:  # If it's a numerical feature
-        user_input[feature] = st.number_input(f"Enter {feature}", value=0.0)
+
+# Numerical inputs (DPD, Credit Expiration)
+for feature in numerical_features:
+    user_input[feature] = st.number_input(f"Enter {feature}", value=0, min_value=0)
+
+# Binary features (Yes/No)
+for feature in binary_features:
+    user_input[feature] = st.selectbox(f"{feature} (Yes/No)", ["Yes", "No"])
+    user_input[feature] = 1 if user_input[feature] == "Yes" else 0  # Convert to 1/0
+
+# Stage as Last Month (Dropdown 1, 2, 3)
+user_input[stage_feature] = st.selectbox("Stage As Last Month", [1, 2, 3])
 
 # Convert input to DataFrame
 input_df = pd.DataFrame([user_input])
 
-# Apply encoding & scaling
-for col, encoder in label_encoders.items():
-    input_df[col] = encoder.transform(input_df[col])
-
-input_df[feature_names] = scaler.transform(input_df[feature_names])
+# Apply scaling
+input_df[numerical_features] = scaler.transform(input_df[numerical_features])
 
 # Predict when user clicks button
 if st.button("Predict"):
@@ -43,5 +46,6 @@ if st.button("Predict"):
     st.success(f"Predicted Stage: {predicted_stage}")
 
 
+
 if __name__ == "__main__":
     main()