AgaMiko ppezik commited on
Commit
b9dbe29
1 Parent(s): d0f8cce

Update README.md (#1)

Browse files

- Update README.md (8abeb0f63759406647695030198202f4e4c39bf2)


Co-authored-by: Piotr Pezik <ppezik@users.noreply.huggingface.co>

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -91,7 +91,7 @@ inputs = ["Christina Katrakis, who spoke to the BBC from Vorokhta in western Ukr
91
  for sample in inputs:
92
  input_sequences = [task_prefix + sample]
93
  input_ids = tokenizer(input_sequences, return_tensors='pt', truncation=True).input_ids
94
- output = model.generate(input_ids, no_repeat_ngram_size=3, num_beams=4)
95
  predicted = tokenizer.decode(output[0], skip_special_tokens=True)
96
  print(sample, "\n --->", predicted)
97
 
 
91
  for sample in inputs:
92
  input_sequences = [task_prefix + sample]
93
  input_ids = tokenizer(input_sequences, return_tensors='pt', truncation=True).input_ids
94
+ output = vlt5.generate(input_ids, no_repeat_ngram_size=3, num_beams=4)
95
  predicted = tokenizer.decode(output[0], skip_special_tokens=True)
96
  print(sample, "\n --->", predicted)
97