T5-Summarization / src /models /predict_model.py
Gagan Bhatia
Update predict_model.py
1ebd2ab
raw
history blame
No virus
234 Bytes
from .model import Summarization
def predict_model(text):
"""
Predict the summary of the given text.
"""
model = Summarization()
model.load_model()
pre_summary = model.predict(text)
return pre_summary