Update app.py

app.py

@@ -55,10 +55,10 @@ if st.button("Предсказать цену"):
 
 
     # Расчет важности фичей с использованием SHAP
-    explainer = shap.TreeExplainer(final_model)
+    explainer = shap.Explainer(final_model)
     shap_values = explainer.shap_values(input_data)
+    shap.force_plot(explainer.expected_value, shap_values, input_data)
 
-    # Отображение SHAP force plot
-    st.write("SHAP Force Plot:")
-    shap.force_plot(explainer.expected_value, shap_values, input_data, matplotlib=True, show=False)
-    st.pyplot()
+    # Показываем график
+    fig, ax = plt.subplots()
+    st.pyplot(fig)