Update README.md (#1)

- Update README.md (8abeb0f63759406647695030198202f4e4c39bf2)


Co-authored-by: Piotr Pezik <[email protected]>

README.md

@@ -91,7 +91,7 @@ inputs = ["Christina Katrakis, who spoke to the BBC from Vorokhta in western Ukr
 for sample in inputs:
        	input_sequences = [task_prefix + sample]
         input_ids = tokenizer(input_sequences, return_tensors='pt', truncation=True).input_ids
-        output = model.generate(input_ids, no_repeat_ngram_size=3, num_beams=4)
+        output = vlt5.generate(input_ids, no_repeat_ngram_size=3, num_beams=4)
         predicted = tokenizer.decode(output[0], skip_special_tokens=True)
         print(sample, "\n --->", predicted)