Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -6,10 +6,10 @@ from joblib import load
|
|
6 |
model = load('loandefaulter.joblib')
|
7 |
scaler = load('scaler.joblib')
|
8 |
|
9 |
-
# Define numerical features for scaling
|
10 |
num_features = [
|
11 |
'loan_amnt', 'int_rate', 'installment', 'annual_inc', 'dti',
|
12 |
-
'revol_bal', 'revol_util', 'total_acc', 'mort_acc'
|
13 |
]
|
14 |
|
15 |
# Create the Streamlit app
|
@@ -53,15 +53,15 @@ input_data = pd.DataFrame({
|
|
53 |
'revol_bal': [revol_bal],
|
54 |
'revol_util': [revol_util],
|
55 |
'total_acc': [total_acc],
|
56 |
-
'mort_acc': [mort_acc]
|
57 |
-
'cibil_score': [cibil_score]
|
58 |
})
|
59 |
|
60 |
# Scale the numerical features that were used to fit the scaler
|
61 |
input_data[num_features] = scaler.transform(input_data[num_features])
|
62 |
|
63 |
-
# Add the
|
64 |
input_data['loan_amnt_by_income'] = [loan_amnt_by_income]
|
|
|
65 |
|
66 |
# Predict using the model
|
67 |
if st.button('Predict'):
|
@@ -71,5 +71,3 @@ if st.button('Predict'):
|
|
71 |
st.markdown(f"""
|
72 |
<div style="font-size: 24px; color: {color}; font-weight: bold;">Prediction: {result}</div>
|
73 |
""", unsafe_allow_html=True)
|
74 |
-
|
75 |
-
|
|
|
6 |
model = load('loandefaulter.joblib')
|
7 |
scaler = load('scaler.joblib')
|
8 |
|
9 |
+
# Define numerical features for scaling (only those that were used during training)
|
10 |
num_features = [
|
11 |
'loan_amnt', 'int_rate', 'installment', 'annual_inc', 'dti',
|
12 |
+
'revol_bal', 'revol_util', 'total_acc', 'mort_acc'
|
13 |
]
|
14 |
|
15 |
# Create the Streamlit app
|
|
|
53 |
'revol_bal': [revol_bal],
|
54 |
'revol_util': [revol_util],
|
55 |
'total_acc': [total_acc],
|
56 |
+
'mort_acc': [mort_acc]
|
|
|
57 |
})
|
58 |
|
59 |
# Scale the numerical features that were used to fit the scaler
|
60 |
input_data[num_features] = scaler.transform(input_data[num_features])
|
61 |
|
62 |
+
# Add the additional feature (not part of scaling)
|
63 |
input_data['loan_amnt_by_income'] = [loan_amnt_by_income]
|
64 |
+
input_data['cibil_score'] = cibil_score # Add cibil_score directly without scaling
|
65 |
|
66 |
# Predict using the model
|
67 |
if st.button('Predict'):
|
|
|
71 |
st.markdown(f"""
|
72 |
<div style="font-size: 24px; color: {color}; font-weight: bold;">Prediction: {result}</div>
|
73 |
""", unsafe_allow_html=True)
|
|
|
|