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']
|