quyip commited on
Commit
4c74009
·
1 Parent(s): da7746d
Files changed (1) hide show
  1. utils/summary_utils.py +2 -2
utils/summary_utils.py CHANGED
@@ -19,7 +19,7 @@ def summarize(id: str, text: str):
19
  return {
20
  "ver": AiSummaryVersion
21
  }
22
- summary = get_summarization(text) if len(text) > 2000 else text
23
  translated = get_en_translation(summary)
24
  tags = get_tags(translated, id)
25
  tags = filter_tags(tags)
@@ -39,7 +39,7 @@ def get_summarization(text: str):
39
  print('XXXXXXXXXXXXXXXXXXXXX1')
40
  print(text)
41
  # Max / Min number of words
42
- result = summarization_pipeline(text, max_length=300, min_length=50, do_sample=False)
43
  print('XXXXXXXXXXXXXXXXXXXXX2')
44
  print(result)
45
  return result[0]['summary_text'] if isinstance(result, list) else result['summary_text']
 
19
  return {
20
  "ver": AiSummaryVersion
21
  }
22
+ summary = get_summarization(text) if len(text) > 3000 else text
23
  translated = get_en_translation(summary)
24
  tags = get_tags(translated, id)
25
  tags = filter_tags(tags)
 
39
  print('XXXXXXXXXXXXXXXXXXXXX1')
40
  print(text)
41
  # Max / Min number of words
42
+ result = summarization_pipeline(text, max_length=500, min_length=200, do_sample=False)
43
  print('XXXXXXXXXXXXXXXXXXXXX2')
44
  print(result)
45
  return result[0]['summary_text'] if isinstance(result, list) else result['summary_text']