varl42 commited on
Commit
830dbbc
·
1 Parent(s): 10ddee6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def summarize_text(text):
34
  inputs = tokenizer(abstract, return_tensors="pt", truncation=True)
35
 
36
  # Generate summary
37
- summary_ids = model.generate(inputs['input_ids'], num_beams=4, max_length=45, min_length=25, early_stopping=True)
38
  summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
39
 
40
  return summary
 
34
  inputs = tokenizer(abstract, return_tensors="pt", truncation=True)
35
 
36
  # Generate summary
37
+ summary_ids = model.generate(inputs['input_ids'], num_beams=4, max_length=45, min_length=30, early_stopping=True)
38
  summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
39
 
40
  return summary