Update app.py

app.py

@@ -40,12 +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(bbox_inches='tight')
-
+st.pyplot(fig, bbox_inches='tight')
+fig2, ax2 = plt.subplots()
 shap.plots.beeswarm(shap_values, show=False)
-st.pyplot(bbox_inches='tight')
-
+st.pyplot(fig2, bbox_inches='tight')
+fig3, ax3 = plt.subplots()
 shap.plots.waterfall(shap_values[0])
-st.pyplot(bbox_inches='tight')
+st.pyplot(fig3, bbox_inches='tight')