Update pages/10_Decision_Tree.py

pages/10_Decision_Tree.py

@@ -129,13 +129,13 @@ elif section == "Visualize Example Tree":
     clf = tree.DecisionTreeClassifier(max_depth=3)
     clf = clf.fit(iris.data, iris.target)
     dot_data = tree.export_graphviz(clf, out_file=None, 
-                                    feature_names=iris.feature_names,
-                                    class_names=iris.target_names,
-                                    filled=True, rounded=True,
-                                    special_characters=True)
-    graph = graphviz.source(dot_data)
+                                feature_names=iris.feature_names,
+                                class_names=iris.target_names,
+                                filled=True, rounded=True,
+                                special_characters=True)
     st.graphviz_chart(dot_data)
 
+
 elif section == "Try with Iris Data":
     st.header("🌸 Try with Iris Dataset")
     iris = load_iris()