Add more stop words to summarized

This commit is contained in:
SillyLossy
2023-04-21 01:11:25 +03:00
parent d3014d8545
commit b0a36f4aaf

View File

@@ -38,7 +38,25 @@ DEFAULT_SUMMARIZE_PARAMS = {
'max_length': 500,
'min_length': 200,
'length_penalty': 1.5,
'bad_words': ["\n", '"', "*", "[", "]", "{", "}", ":", "(", ")", "<", ">", "Â"]
'bad_words': [
"\n",
'"',
"*",
"[",
"]",
"{",
"}",
":",
"(",
")",
"<",
">",
"Â",
"The text ends",
"The story ends",
"The text is",
"The story is",
]
}
class SplitArgs(argparse.Action):