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
|