Update README.md

README.md

@@ -21,7 +21,7 @@ model = MT5ForConditionalGeneration.from_pretrained('iliemihai/mt5-base-romanian
 tokenizer = T5Tokenizer.from_pretrained('iliemihai/mt5-base-romanian-diacritics')
 input_text = "A inceput sa ii taie un fir de par, iar fata sta in fata, tine camasa de in in mana si canta nota SI."
 inputs = tokenizer(input_text, max_length=256, truncation=True, return_tensors="pt")
-outputs = model.generate(input_ids=inputs["input_ids"], attention_mask=input_text["attention_mask"])
+outputs = model.generate(input_ids=inputs["input_ids"], attention_mask=inputs["attention_mask"])
 output = tokenizer.decode(outputs[0], skip_special_tokens=True)
 print(output)  # this will print "A început să îi taie un fir de păr, iar fata stă în față, ține cămașa de in în mână și cântă nota SI" 
 ```