ppezik commited on
Commit
8abeb0f
1 Parent(s): d0f8cce

Update README.md

Browse files

Fix model variable name

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