gabrielwithhappy commited on
Commit
1eb5133
1 Parent(s): d162cd8

remove top_p

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def get_summary(input_text):
18
  input_ids = tokenizer.encode(input_text, return_tensors="pt")
19
  summary_text_ids = summary_model.generate(
20
  length_penalty=1,
21
- top_p=0.9,
22
  max_length=128,
23
  min_length=12,
24
  num_beams=4,
 
18
  input_ids = tokenizer.encode(input_text, return_tensors="pt")
19
  summary_text_ids = summary_model.generate(
20
  length_penalty=1,
21
+ # top_p=0.9,
22
  max_length=128,
23
  min_length=12,
24
  num_beams=4,