quyip commited on
Commit
652d52e
·
1 Parent(s): 3742ff7
Files changed (1) hide show
  1. utils/summary_utils.py +1 -1
utils/summary_utils.py CHANGED
@@ -38,7 +38,7 @@ def get_summarization(text: str):
38
  try:
39
  print('XXXXXXXXXXXXXXXXXXXXX1')
40
  print(text)
41
- result = summarization_pipeline(text, max_length=50000, min_length=2000, 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
+ 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']