Update README.md
Browse filesFix model variable name
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 =
|
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 |
|