Gagan Bhatia commited on
Commit
1ebd2ab
1 Parent(s): 8b14993

Update predict_model.py

Browse files
Files changed (1) hide show
  1. src/models/predict_model.py +6 -0
src/models/predict_model.py CHANGED
@@ -3,3 +3,9 @@ from .model import Summarization
3
  def predict_model(text):
4
  """
5
  Predict the summary of the given text.
 
 
 
 
 
 
 
3
  def predict_model(text):
4
  """
5
  Predict the summary of the given text.
6
+ """
7
+ model = Summarization()
8
+ model.load_model()
9
+ pre_summary = model.predict(text)
10
+ return pre_summary
11
+