Update app.py

app.py

@@ -87,8 +87,6 @@ def load_model_and_data():
 
     return model, X, y
     
-   
-
 import streamlit as st
 import pandas as pd
 import numpy as np
@@ -96,7 +94,7 @@ import matplotlib.pyplot as plt
 from sklearn.tree import plot_tree, export_text
 import seaborn as sns
 from sklearn.preprocessing import LabelEncoder
-from dtreeviz.trees import dtreeviz
+from dtreeviz import trees
 
 
 def app():
@@ -155,7 +153,7 @@ def app():
             st.text(tree_text)
         else:
             # Création de la visualisation dtreeviz
-            viz = dtreeviz(
+            viz = trees.dtreeviz(
                 model,
                 X,
                 y,
@@ -165,9 +163,10 @@ def app():
                 max_depth=max_depth
             )
             
-            # Conversion en PNG pour Streamlit
-            png_data = viz.save_png()
-            st.image(png_data, use_column_width=True)
+            # Sauvegarde temporaire et affichage
+            st.set_option('deprecation.showPyplotGlobalUse', False)
+            fig = viz.view()
+            st.pyplot(fig)
     
     # Analyse de cohortes
     elif page == "Analyse de cohortes":
@@ -254,4 +253,4 @@ def app():
                 st.write(rule)
 
 if __name__ == "__main__":
-    app()
+    app()