Gagan Bhatia commited on
Commit
0c204b8
1 Parent(s): 241e6ac

Update predict_model.py

Browse files
Files changed (1) hide show
  1. src/models/predict_model.py +1 -1
src/models/predict_model.py CHANGED
@@ -17,7 +17,7 @@ def predict_model(text):
17
  pre_summary = model.predict(text)
18
  return pre_summary
19
 
20
-
21
  if __name__ == '__main__':
22
  text = make_dataset(split="test")['input_text'][0]
23
  pre_summary = predict_model(text)
 
17
  pre_summary = model.predict(text)
18
  return pre_summary
19
 
20
+
21
  if __name__ == '__main__':
22
  text = make_dataset(split="test")['input_text'][0]
23
  pre_summary = predict_model(text)