Kavin0003 commited on
Commit
185e4f4
·
1 Parent(s): 7adddb2

⬆️ Updated app.py with max_tokens

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def correct_grammar(text):
19
  with torch.no_grad():
20
  outputs = model.generate(
21
  **inputs,
22
- max_length=128,
23
  num_beams=5,
24
  early_stopping=True,
25
  temperature=0.7,
 
19
  with torch.no_grad():
20
  outputs = model.generate(
21
  **inputs,
22
+ max_length=1024,
23
  num_beams=5,
24
  early_stopping=True,
25
  temperature=0.7,