Update README.md

README.md

@@ -33,8 +33,8 @@ input_sentences = ["Yo va al tienda.", "Espero que tú ganas."]
 
 tokenized_text = tokenizer(input_sentences, return_tensors="pt")
 
-input_ids = source_enc["input_ids"].squeeze()
-attention_mask = source_enc["attention_mask"].squeeze()
+input_ids = tokenized_text["input_ids"].squeeze()
+attention_mask = tokenized_text["attention_mask"].squeeze()
 
 outputs = model.generate(input_ids=input_ids, attention_mask=attention_mask)