google
/

bigbird-pegasus-large-pubmed

bigbird_pegasus

text2text-generation

Inference Endpoints

Model card Files Files and versions Community

vasudevgupta commited on May 4, 2021

Commit

34117d8

•

1 Parent(s): b97fe84

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -38,7 +38,7 @@ model = BigBirdPegasusForConditionalGeneration.from_pretrained("google/bigbird-p
 text = "Replace me by any text you'd like."
 inputs = tokenizer(text, return_tensors='pt')
-prediction = model.generate(input_ids=input_ids, num_beams=5, length_penalty=0.8)
 prediction = tokenizer.batch_decode(prediction)
 ```

 text = "Replace me by any text you'd like."
 inputs = tokenizer(text, return_tensors='pt')
+prediction = model.generate(**inputs)
 prediction = tokenizer.batch_decode(prediction)
 ```