Update __init__.py

src/models/__init__.py

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