Update app.py

app.py

@@ -52,9 +52,11 @@ def st_shap(plot, height=None):
 #The button
 if st.button("Calculate Your House Price!"):
     results = predictor.make_a_prediction(data)
-    results_f = '{0:.2f}'.format(results[0])
-    st.subheader('$' + results_f)
-    st.write("Below are the SHAP graphs which can be used to explain the decisions made by the model using the features provided.")
+    opt_f = '{0:.2f}'.format(results[0][0])
+    unopt_f = '{0:.2f}'.format(results[0][1])
+    st.subheader('Optimized: $' + opt_f)
+    st.subheader('Unoptimized: $' + unopt_f)
+    st.write("Below are the SHAP graphs which can be used to explain the decisions made by the model using the features provided for the *optimized* results.")
     force_plot = predictor.create_shap_models(data)
     st_shap(force_plot)