Spaces:
Runtime error
Runtime error
quyip
commited on
Commit
·
06d2291
1
Parent(s):
c775a48
fix
Browse files- 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('
|
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 []
|