Update README.md
Browse files
README.md
CHANGED
@@ -47,7 +47,7 @@ def tag(text: str):
|
|
47 |
early_stopping=True)
|
48 |
|
49 |
output = tokenizer.decode(tags_ids[0], skip_special_tokens=True)
|
50 |
-
return output.split(',')
|
51 |
|
52 |
tags = tag(article)
|
53 |
print(tags)
|
|
|
47 |
early_stopping=True)
|
48 |
|
49 |
output = tokenizer.decode(tags_ids[0], skip_special_tokens=True)
|
50 |
+
return output.split(', ')
|
51 |
|
52 |
tags = tag(article)
|
53 |
print(tags)
|