Update __init__.py

src/models/__init__.py

@@ -1,3 +1,3 @@
-from .train_model import train_model
 from .predict_model import predict_model
 from .evaluate_model import evaluate_modelfrom .train_model import train_model  # noqa: F401
+from .model import Summarization  # noqa: F401