varl42 commited on
Commit
4469f9e
·
1 Parent(s): cf073d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -32,15 +32,15 @@ def summarize_text(text):
32
 
33
  # Tokenize abstract
34
  inputs = tokenizer(abstract,
35
- max_length=1024,
36
  return_tensors="pt",
37
  truncation=True)
38
 
39
  # Generate summary
40
  summary_ids = model.generate(inputs['input_ids'],
41
  num_beams=3,
42
- max_length=40,
43
- min_length=25,
44
  do_sample=False,
45
  early_stopping=True)
46
  summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
 
32
 
33
  # Tokenize abstract
34
  inputs = tokenizer(abstract,
35
+ #max_length=1024,
36
  return_tensors="pt",
37
  truncation=True)
38
 
39
  # Generate summary
40
  summary_ids = model.generate(inputs['input_ids'],
41
  num_beams=3,
42
+ max_length=50,
43
+ min_length=30,
44
  do_sample=False,
45
  early_stopping=True)
46
  summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)