quyip commited on
Commit
6e7cb96
·
1 Parent(s): 090f2d4
Files changed (1) hide show
  1. utils/summary_utils.py +8 -5
utils/summary_utils.py CHANGED
@@ -68,11 +68,14 @@ def get_tags(text: str):
68
  if text is None:
69
  return []
70
  try:
71
- tags1 = [tag['label'] for tag in tag_gen_pipe_1(text) if tag['score'] >= MinTagScore]
72
- tags2 = [tag['label'] for tag in tag_gen_pipe_2(text) if tag['score'] >= MinTagScore]
73
- tags3 = [tag['label'] for tag in tag_gen_pipe_3(text) if tag['score'] >= MinTagScore]
74
- print(tags1, tags2, tags3)
75
- return tags1 + tags2 + tags3
 
 
 
76
  except:
77
  return []
78
 
 
68
  if text is None:
69
  return []
70
  try:
71
+ print(tag_gen_pipe_1(text))
72
+ print(tag_gen_pipe_2(text))
73
+ print(tag_gen_pipe_3(text))
74
+ # tags1 = [tag['label'] for tag in tag_gen_pipe_1(text) if tag['score'] >= MinTagScore]
75
+ # tags2 = [tag['label'] for tag in tag_gen_pipe_2(text) if tag['score'] >= MinTagScore]
76
+ # tags3 = [tag['label'] for tag in tag_gen_pipe_3(text) if tag['score'] >= MinTagScore]
77
+ # print(tags1, tags2, tags3)
78
+ return []
79
  except:
80
  return []
81