Update __init__.py

src/models/__init__.py

@@ -1,2 +1,3 @@
 from .model import Summarization
 from .train_model import train_model
+from .predict_model import predict_model