Update app.py

app.py

@@ -46,13 +46,13 @@ if st.sidebar.button('Run Algorithm'):
                                 min_impurity_decrease=min_impurity_decrease)
     clf.fit(X_train, y_train)
     y_pred = clf.predict(X_test)
-    viz_model = dtreeviz.model(clf,
+    viz_model = dtreeviz(clf,
                            X_train=X, y_train=y,
                            feature_names=iris.feature_names,
                            target_name='iris',
                            class_names=iris.target_names)
 
-    v = viz_model.view()     # render as SVG into internal object 
+    st.graphviz_chart(viz_model)    # render as SVG into internal object 
     # v.show()                 # pop up window
     # v.save("/tmp/iris.svg")
     def svg_write(svg, center=True):