Spaces:
Runtime error
Runtime error
quyip
commited on
Commit
·
2c41d85
1
Parent(s):
fd9b289
fix
Browse files- utils/summary_utils.py +3 -3
utils/summary_utils.py
CHANGED
|
@@ -19,8 +19,8 @@ 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(
|
| 24 |
tags = get_tags(translated, id)
|
| 25 |
tags = filter_tags(tags)
|
| 26 |
tags = sorted(list(set(tags)))
|
|
@@ -71,8 +71,8 @@ def get_tags(text: str, id: str):
|
|
| 71 |
tags3 = [tag['label'] for tag in tag_gen_pipe_3(text) if tag['score'] >= MinTagScore]
|
| 72 |
print('XXXXXXXXXXXXXXXXXXXXX')
|
| 73 |
print(id)
|
| 74 |
-
print(tags1, tags2, tags3)
|
| 75 |
print(text)
|
|
|
|
| 76 |
return tags1 + tags2 + tags3
|
| 77 |
except:
|
| 78 |
return []
|
|
|
|
| 19 |
return {
|
| 20 |
"ver": AiSummaryVersion
|
| 21 |
}
|
| 22 |
+
summary = get_summarization(text) if len(text) > 2000 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)))
|
|
|
|
| 71 |
tags3 = [tag['label'] for tag in tag_gen_pipe_3(text) if tag['score'] >= MinTagScore]
|
| 72 |
print('XXXXXXXXXXXXXXXXXXXXX')
|
| 73 |
print(id)
|
|
|
|
| 74 |
print(text)
|
| 75 |
+
print(tags1, tags2, tags3)
|
| 76 |
return tags1 + tags2 + tags3
|
| 77 |
except:
|
| 78 |
return []
|