Update __init__.py

src/__init__.py

@@ -1,2 +1,3 @@
 from src.models.evaluate_model import evaluate_model
 from src.models.predict_model import predict_model
+from src.models.train_model import train_model