Update __init__.py

src/models/__init__.py

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