Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import pandas as pd
 import pickle
-from utils import create_new_features, normalize, init_new_pred
+from utils import create_new_features, normalize, bucketize, init_new_pred
 
 with open('./trained_model.pkl', 'rb') as file:
     model = pickle.load(file)
@@ -65,6 +65,7 @@ with col1:
     # Process the prediction
     new_pred = pd.DataFrame([new_pred])
     new_pred = create_new_features(new_pred)
+    new_pred = bucketize(new_pred)
     new_pred = normalize(new_pred)
 
     # Predict the price