Update main.py

main.py

@@ -96,7 +96,7 @@ def main():
                 
                 # Render and prepare the diagram for download
                 filename = f"{row['Student']}_decision_tree"
-                filepath = download_tree_diagram(tree_diagram, filename)
+                filepath = visualization.download_tree_diagram(tree_diagram, filename)
                 
                 # Provide a download button
                 with open(filepath, "rb") as file: