eloi-goncalves commited on
Commit
03113eb
·
1 Parent(s): 0c4b135

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ mdl = PegasusForConditionalGeneration.from_pretrained(mdl_name)
33
 
34
  def summarize(text):
35
  tokens = pegasus_tkn(text, truncation=True, padding="longest", return_tensors="pt")
36
- txt_summary = mdl.generate(**tokens)
37
  response = pegasus_tkn.batch_decode(txt_summary, skip_special_tokens=True)
38
  return response
39
  txt=grad.Textbox(lines=10, label="English", placeholder="English Text here")
 
33
 
34
  def summarize(text):
35
  tokens = pegasus_tkn(text, truncation=True, padding="longest", return_tensors="pt")
36
+ txt_summary = mdl.generate(**tokens, num_return_sequences=5, max_length=200, temperature=1.5,num_beams=10)
37
  response = pegasus_tkn.batch_decode(txt_summary, skip_special_tokens=True)
38
  return response
39
  txt=grad.Textbox(lines=10, label="English", placeholder="English Text here")