quyip commited on
Commit
3784e1c
·
1 Parent(s): 6e7cb96
Files changed (1) hide show
  1. utils/summary_utils.py +6 -3
utils/summary_utils.py CHANGED
@@ -71,9 +71,12 @@ def get_tags(text: str):
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:
 
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)
78
+ print(tags2)
79
+ print(tags3)
80
  # print(tags1, tags2, tags3)
81
  return []
82
  except: