Spaces:
Runtime error
Runtime error
quyip
commited on
Commit
·
30e268a
1
Parent(s):
652d52e
fix
Browse files- utils/summary_utils.py +2 -1
utils/summary_utils.py
CHANGED
|
@@ -38,7 +38,8 @@ def get_summarization(text: str):
|
|
| 38 |
try:
|
| 39 |
print('XXXXXXXXXXXXXXXXXXXXX1')
|
| 40 |
print(text)
|
| 41 |
-
|
|
|
|
| 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 |
+
# Max / Min number of words
|
| 42 |
+
result = summarization_pipeline(text, max_length=300, min_length=100, do_sample=False)
|
| 43 |
print('XXXXXXXXXXXXXXXXXXXXX2')
|
| 44 |
print(result)
|
| 45 |
return result[0]['summary_text'] if isinstance(result, list) else result['summary_text']
|