Update README.md
Browse files
README.md
CHANGED
|
@@ -38,6 +38,7 @@ La scoperta gli valse il Nobel per la Medicina. È uscita in queste settimane pe
|
|
| 38 |
def tag(text: str):
|
| 39 |
""" Generates tags from given text """
|
| 40 |
text = text.strip().replace('\n', '')
|
|
|
|
| 41 |
tokenized_text = tokenizer.encode(text, return_tensors="pt")
|
| 42 |
|
| 43 |
tags_ids = model.generate(tokenized_text,
|
|
|
|
| 38 |
def tag(text: str):
|
| 39 |
""" Generates tags from given text """
|
| 40 |
text = text.strip().replace('\n', '')
|
| 41 |
+
text = 'summarize: ' + text
|
| 42 |
tokenized_text = tokenizer.encode(text, return_tensors="pt")
|
| 43 |
|
| 44 |
tags_ids = model.generate(tokenized_text,
|