Voicelab
/

vlt5-base-keywords

Text2Text Generation

keywords-generation

text-classifiation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

madhvendra21 commited on Aug 10, 2023

Commit

dc9b67f

•

1 Parent(s): 9d04f1d

Update README.md

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)