updates

src/__init__.py

@@ -0,0 +1,8 @@
+from src.models.evaluate_model import evaluate_model
+from src.models.predict_model import predict_model
+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
+import os, sys; sys.path.append(os.path.dirname(os.path.realpath(__file__)))

src/models/__init__.py

@@ -1 +1,4 @@
 from .model import Summarization
+from .train_model import train_model
+from .predict_model import predict_model
+from .evaluate_model import evaluate_model

src/visualization/visualize.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from src.models.predict_model import predict_model
+from models import predict_model
 
 
 def visualize():