Spaces:
Runtime error
Runtime error
quyip
commited on
Commit
·
652d52e
1
Parent(s):
3742ff7
fix
Browse files- utils/summary_utils.py +1 -1
utils/summary_utils.py
CHANGED
@@ -38,7 +38,7 @@ def get_summarization(text: str):
|
|
38 |
try:
|
39 |
print('XXXXXXXXXXXXXXXXXXXXX1')
|
40 |
print(text)
|
41 |
-
result = summarization_pipeline(text, max_length=
|
42 |
print('XXXXXXXXXXXXXXXXXXXXX2')
|
43 |
print(result)
|
44 |
return result[0]['summary_text'] if isinstance(result, list) else result['summary_text']
|
|
|
38 |
try:
|
39 |
print('XXXXXXXXXXXXXXXXXXXXX1')
|
40 |
print(text)
|
41 |
+
result = summarization_pipeline(text, max_length=2000, min_length=500, do_sample=False)
|
42 |
print('XXXXXXXXXXXXXXXXXXXXX2')
|
43 |
print(result)
|
44 |
return result[0]['summary_text'] if isinstance(result, list) else result['summary_text']
|