Update app.py

app.py

@@ -56,10 +56,11 @@ if st.button("Предсказать цену"):
 
 
     # Расчет важности фичей с использованием SHAP
-    explainer = shap.Explainer(final_model)
+    explainer = shap.TreeExplainer(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:")
     fig, ax = plt.subplots()
+    shap.force_plot(explainer.expected_value, shap_values, input_data, matplotlib=True, show=False, ax=ax)
     st.pyplot(fig)