Voicelab
/

vlt5-base-keywords

Text2Text Generation

keywords-generation

text-classifiation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Community

Update README.md

#4

by madhvendra21 - opened Aug 10, 2023

base: refs/heads/main

←

from: refs/pr/4

Discussion Files changed

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -95,7 +95,7 @@ for sample in inputs:
     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)
     predicted = tokenizer.decode(output[0], skip_special_tokens=True)
     print(sample, "\n --->", predicted)

     input_ids = tokenizer(
         input_sequences, return_tensors="pt", truncation=True
     ).input_ids
+    output = model.generate(input_ids, no_repeat_ngram_size=3, num_beams=20)
     predicted = tokenizer.decode(output[0], skip_special_tokens=True)
     print(sample, "\n --->", predicted)