Update app.py

app.py

@@ -118,8 +118,6 @@ lgbm_base = pickle.load(open('lgbm_base.pkl', 'rb'))
 lgbm_opt = pickle.load(open('lgbm_optimized.pkl', 'rb'))
 xgb = pickle.load(open('xgb_model.pkl', 'rb'))
 
-st.write(lgbm_base)
-
 y_pred_xgb = xgb.predict(data_df)
 #y_pred_optimized = lgbm_opt.predict(data_df)
 
@@ -154,8 +152,8 @@ if center_button:
 
     col1, col2 = st.columns([3, 3])
 
-    lower_number = "{:,.2f}".format(int(y_pred_xgb.mean().numpy()-1.95*yhat.stddev().numpy()))
-    higher_number = "{:,.2f}".format(int(y_pred_xgb.mean().numpy()+1.95*yhat.stddev().numpy()))
+    lower_number = "{:,.2f}".format(y_pred_xgb[0])
+    higher_number = "{:,.2f}".format(y_pred_xgb[0])
 
     col1, col2, col3 = st.columns(3)