quyip commited on
Commit
60a3519
·
1 Parent(s): 4c74009
Files changed (1) hide show
  1. utils/summary_utils.py +5 -5
utils/summary_utils.py CHANGED
@@ -19,10 +19,14 @@ def summarize(id: str, text: str):
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)
 
26
  tags = sorted(list(set(tags)))
27
 
28
  value = {
@@ -36,12 +40,8 @@ def summarize(id: str, text: str):
36
 
37
  def get_summarization(text: str):
38
  try:
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']
46
  except:
47
  return None
 
19
  return {
20
  "ver": AiSummaryVersion
21
  }
22
+ print('XXXXXXXXXXXXXXXXXXXXX text')
23
  summary = get_summarization(text) if len(text) > 3000 else text
24
+ print('XXXXXXXXXXXXXXXXXXXXX summary')
25
+ print(summary)
26
  translated = get_en_translation(summary)
27
  tags = get_tags(translated, id)
28
  tags = filter_tags(tags)
29
+ print(tags)
30
  tags = sorted(list(set(tags)))
31
 
32
  value = {
 
40
 
41
  def get_summarization(text: str):
42
  try:
 
 
43
  # Max / Min number of words
44
+ result = summarization_pipeline(text, max_length=500, min_length=100, do_sample=False)
 
 
45
  return result[0]['summary_text'] if isinstance(result, list) else result['summary_text']
46
  except:
47
  return None