quyip commited on
Commit
30e268a
·
1 Parent(s): 652d52e
Files changed (1) hide show
  1. utils/summary_utils.py +2 -1
utils/summary_utils.py CHANGED
@@ -38,7 +38,8 @@ def get_summarization(text: str):
38
  try:
39
  print('XXXXXXXXXXXXXXXXXXXXX1')
40
  print(text)
41
- result = summarization_pipeline(text, max_length=2000, min_length=500, do_sample=False)
 
42
  print('XXXXXXXXXXXXXXXXXXXXX2')
43
  print(result)
44
  return result[0]['summary_text'] if isinstance(result, list) else result['summary_text']
 
38
  try:
39
  print('XXXXXXXXXXXXXXXXXXXXX1')
40
  print(text)
41
+ # Max / Min number of words
42
+ result = summarization_pipeline(text, max_length=300, min_length=100, do_sample=False)
43
  print('XXXXXXXXXXXXXXXXXXXXX2')
44
  print(result)
45
  return result[0]['summary_text'] if isinstance(result, list) else result['summary_text']