Spaces:
Runtime error
Runtime error
quyip
commited on
Commit
·
4c74009
1
Parent(s):
da7746d
fix
Browse files- utils/summary_utils.py +2 -2
utils/summary_utils.py
CHANGED
@@ -19,7 +19,7 @@ def summarize(id: str, text: str):
|
|
19 |
return {
|
20 |
"ver": AiSummaryVersion
|
21 |
}
|
22 |
-
summary = get_summarization(text) if len(text) >
|
23 |
translated = get_en_translation(summary)
|
24 |
tags = get_tags(translated, id)
|
25 |
tags = filter_tags(tags)
|
@@ -39,7 +39,7 @@ def get_summarization(text: str):
|
|
39 |
print('XXXXXXXXXXXXXXXXXXXXX1')
|
40 |
print(text)
|
41 |
# Max / Min number of words
|
42 |
-
result = summarization_pipeline(text, max_length=
|
43 |
print('XXXXXXXXXXXXXXXXXXXXX2')
|
44 |
print(result)
|
45 |
return result[0]['summary_text'] if isinstance(result, list) else result['summary_text']
|
|
|
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)
|
|
|
39 |
print('XXXXXXXXXXXXXXXXXXXXX1')
|
40 |
print(text)
|
41 |
# Max / Min number of words
|
42 |
+
result = summarization_pipeline(text, max_length=500, min_length=200, do_sample=False)
|
43 |
print('XXXXXXXXXXXXXXXXXXXXX2')
|
44 |
print(result)
|
45 |
return result[0]['summary_text'] if isinstance(result, list) else result['summary_text']
|