Gagan Bhatia commited on
Commit
ca3e561
1 Parent(s): 81b7588

Update predict_model.py

Browse files
Files changed (1) hide show
  1. src/models/predict_model.py +1 -0
src/models/predict_model.py CHANGED
@@ -14,3 +14,4 @@ def predict_model(text):
14
  if __name__ == '__main__':
15
  text = make_dataset(split="test")['input_text'][0]
16
  pre_summary = predict_model(text)
 
 
14
  if __name__ == '__main__':
15
  text = make_dataset(split="test")['input_text'][0]
16
  pre_summary = predict_model(text)
17
+ print(pre_summary)