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