Update __init__.py

src/__init__.py

@@ -4,3 +4,4 @@ from src.models.train_model import train_model
 from src.models.model import Summarization
 from src.data.make_dataset import make_dataset
 from src.data.process_data import process_data
+from src.visualization.visualize import visualize