Update app.py

app.py

@@ -54,13 +54,14 @@ if st.button("Предсказать цену"):
     # Вывод результатов
     st.write(f"Прогнозируемая цена авто: {predicted_price} руб.")
 
-
+    # Отключение предупреждения PyplotGlobalUseWarning
+    st.set_option('deprecation.showPyplotGlobalUse', False)
+    
     # Расчет важности фичей с использованием SHAP
     explainer = shap.TreeExplainer(final_model)
     shap_values = explainer.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)
-    st.pyplot(fig)
+    shap.force_plot(explainer.expected_value, shap_values[0], input_data, matplotlib=True, show=False)
+    st.pyplot()