Update README.md
Browse files
README.md
CHANGED
|
@@ -31,7 +31,7 @@ model = BartForConditionalGeneration.from_pretrained("SkitCon/gec-spanish-BARTO-
|
|
| 31 |
|
| 32 |
input_sentences = ["Yo va al tienda.", "Espero que tú ganas."]
|
| 33 |
|
| 34 |
-
tokenized_text = tokenizer(input_sentences, return_tensors="pt")
|
| 35 |
|
| 36 |
input_ids = tokenized_text["input_ids"].squeeze()
|
| 37 |
attention_mask = tokenized_text["attention_mask"].squeeze()
|
|
|
|
| 31 |
|
| 32 |
input_sentences = ["Yo va al tienda.", "Espero que tú ganas."]
|
| 33 |
|
| 34 |
+
tokenized_text = tokenizer(input_sentences, max_length=128, padding="max_length", truncation=True, return_tensors="pt")
|
| 35 |
|
| 36 |
input_ids = tokenized_text["input_ids"].squeeze()
|
| 37 |
attention_mask = tokenized_text["attention_mask"].squeeze()
|