Update app.py

app.py

@@ -40,9 +40,12 @@ prediction = model.predict(input_df)
 st.write(f"The estimated house price range is ${prediction[0]:,.2f}")
 explainer = shap.Explainer(model, X_train)
 shap_values = explainer(X_test)
-fig, ax = plt.subplots()
+
 shap.summary_plot(shap_values, X_test, plot_type="bar", show=False)
-st.pyplot(fig, bbox_inches='tight')
-fig2, ax2 = plt.subplots()
+st.pyplot(bbox_inches='tight')
+
 shap.plots.beeswarm(shap_values, show=False)
-st.pyplot(fig2, bbox_inches='tight')
+st.pyplot(bbox_inches='tight')
+
+shap.plots.waterfall(shap_values[0])
+st.pyplot(bbox_inches='tight')