Rahmat82 commited on
Commit
9347162
·
verified ·
1 Parent(s): e0fcb2d

set max_token to 100

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,9 +15,9 @@ async def summarize(text, retries=5):
15
  data = {
16
  "inputs": text,
17
  "options": {
18
- "max_new_tokens": 60,
19
  "min_length": 10,
20
- "max_length": 300,
21
  "length_penalty": 2.0,
22
  "num_beams": 3,
23
  "early_stopping": True
 
15
  data = {
16
  "inputs": text,
17
  "options": {
18
+ "max_new_tokens": 100,
19
  "min_length": 10,
20
+ "max_length": 500,
21
  "length_penalty": 2.0,
22
  "num_beams": 3,
23
  "early_stopping": True