quyip commited on
Commit
2c41d85
·
1 Parent(s): fd9b289
Files changed (1) hide show
  1. utils/summary_utils.py +3 -3
utils/summary_utils.py CHANGED
@@ -19,8 +19,8 @@ def summarize(id: str, text: str):
19
  return {
20
  "ver": AiSummaryVersion
21
  }
22
- summary = get_summarization(text) if len(text) > 100 else text
23
- translated = get_en_translation(text)
24
  tags = get_tags(translated, id)
25
  tags = filter_tags(tags)
26
  tags = sorted(list(set(tags)))
@@ -71,8 +71,8 @@ def get_tags(text: str, id: str):
71
  tags3 = [tag['label'] for tag in tag_gen_pipe_3(text) if tag['score'] >= MinTagScore]
72
  print('XXXXXXXXXXXXXXXXXXXXX')
73
  print(id)
74
- print(tags1, tags2, tags3)
75
  print(text)
 
76
  return tags1 + tags2 + tags3
77
  except:
78
  return []
 
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)
26
  tags = sorted(list(set(tags)))
 
71
  tags3 = [tag['label'] for tag in tag_gen_pipe_3(text) if tag['score'] >= MinTagScore]
72
  print('XXXXXXXXXXXXXXXXXXXXX')
73
  print(id)
 
74
  print(text)
75
+ print(tags1, tags2, tags3)
76
  return tags1 + tags2 + tags3
77
  except:
78
  return []