Gregniuki commited on
Commit
3ebdedd
·
verified ·
1 Parent(s): 65d9b39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -165,8 +165,8 @@ def translate_batch(text_input):
165
  with torch.no_grad():
166
  outputs = model.generate(
167
  **inputs,
168
- max_length=768,
169
- num_beams=4,
170
  early_stopping=True
171
  )
172
  # output_ids shape: [batch_size, sequence_length]
 
165
  with torch.no_grad():
166
  outputs = model.generate(
167
  **inputs,
168
+ max_length=1024,
169
+ num_beams=8,
170
  early_stopping=True
171
  )
172
  # output_ids shape: [batch_size, sequence_length]