Update __init__.py

src/models/__init__.py

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