quyip commited on
Commit
06d2291
·
1 Parent(s): c775a48
Files changed (1) hide show
  1. utils/summary_utils.py +2 -4
utils/summary_utils.py CHANGED
@@ -36,9 +36,10 @@ def summarize(id: str, text: str):
36
 
37
  def get_summarization(text: str):
38
  try:
39
- print('XXXXXXXXXXXXXXXXXXXXX')
40
  print(text)
41
  result = summarization_pipeline(text)
 
42
  print(result)
43
  return result[0]['summary_text'] if isinstance(result, list) else result['summary_text']
44
  except:
@@ -72,9 +73,6 @@ def get_tags(text: str, id: str):
72
  tags1 = [tag['label'] for tag in tag_gen_pipe_1(text) if tag['score'] >= MinTagScore]
73
  tags2 = [tag['label'] for tag in tag_gen_pipe_2(text) if tag['score'] >= MinTagScore]
74
  tags3 = [tag['label'] for tag in tag_gen_pipe_3(text) if tag['score'] >= MinTagScore]
75
- print('XXXXXXXXXXXXXXXXXXXXX')
76
- print(id)
77
- print(tags1, tags2, tags3)
78
  return tags1 + tags2 + tags3
79
  except:
80
  return []
 
36
 
37
  def get_summarization(text: str):
38
  try:
39
+ print('XXXXXXXXXXXXXXXXXXXXX1')
40
  print(text)
41
  result = summarization_pipeline(text)
42
+ print('XXXXXXXXXXXXXXXXXXXXX2')
43
  print(result)
44
  return result[0]['summary_text'] if isinstance(result, list) else result['summary_text']
45
  except:
 
73
  tags1 = [tag['label'] for tag in tag_gen_pipe_1(text) if tag['score'] >= MinTagScore]
74
  tags2 = [tag['label'] for tag in tag_gen_pipe_2(text) if tag['score'] >= MinTagScore]
75
  tags3 = [tag['label'] for tag in tag_gen_pipe_3(text) if tag['score'] >= MinTagScore]
 
 
 
76
  return tags1 + tags2 + tags3
77
  except:
78
  return []