Spaces:
Runtime error
Runtime error
quyip
commited on
Commit
·
60a3519
1
Parent(s):
4c74009
fix
Browse files- utils/summary_utils.py +5 -5
utils/summary_utils.py
CHANGED
@@ -19,10 +19,14 @@ def summarize(id: str, text: str):
|
|
19 |
return {
|
20 |
"ver": AiSummaryVersion
|
21 |
}
|
|
|
22 |
summary = get_summarization(text) if len(text) > 3000 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)))
|
27 |
|
28 |
value = {
|
@@ -36,12 +40,8 @@ def summarize(id: str, text: str):
|
|
36 |
|
37 |
def get_summarization(text: str):
|
38 |
try:
|
39 |
-
print('XXXXXXXXXXXXXXXXXXXXX1')
|
40 |
-
print(text)
|
41 |
# Max / Min number of words
|
42 |
-
result = summarization_pipeline(text, max_length=500, min_length=
|
43 |
-
print('XXXXXXXXXXXXXXXXXXXXX2')
|
44 |
-
print(result)
|
45 |
return result[0]['summary_text'] if isinstance(result, list) else result['summary_text']
|
46 |
except:
|
47 |
return None
|
|
|
19 |
return {
|
20 |
"ver": AiSummaryVersion
|
21 |
}
|
22 |
+
print('XXXXXXXXXXXXXXXXXXXXX text')
|
23 |
summary = get_summarization(text) if len(text) > 3000 else text
|
24 |
+
print('XXXXXXXXXXXXXXXXXXXXX summary')
|
25 |
+
print(summary)
|
26 |
translated = get_en_translation(summary)
|
27 |
tags = get_tags(translated, id)
|
28 |
tags = filter_tags(tags)
|
29 |
+
print(tags)
|
30 |
tags = sorted(list(set(tags)))
|
31 |
|
32 |
value = {
|
|
|
40 |
|
41 |
def get_summarization(text: str):
|
42 |
try:
|
|
|
|
|
43 |
# Max / Min number of words
|
44 |
+
result = summarization_pipeline(text, max_length=500, min_length=100, do_sample=False)
|
|
|
|
|
45 |
return result[0]['summary_text'] if isinstance(result, list) else result['summary_text']
|
46 |
except:
|
47 |
return None
|