Enutrof commited on
Commit
6388a03
·
1 Parent(s): 630ddee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,9 +12,9 @@ def load_translator(model_name='Helsinki-NLP/opus-mt-en-fr'):
12
  :return: model
13
  '''
14
  pmodel_args = Seq2SeqArgs()
15
- pmodel_args.max_length = 1024
16
  pmodel_args.length_penalty = 1
17
- pmodel_args.num_beams = 20
18
  pmodel_args.num_return_sequences = 3
19
 
20
  pmodel = Seq2SeqModel(
 
12
  :return: model
13
  '''
14
  pmodel_args = Seq2SeqArgs()
15
+ pmodel_args.max_length = 4096
16
  pmodel_args.length_penalty = 1
17
+ pmodel_args.num_beams = 5
18
  pmodel_args.num_return_sequences = 3
19
 
20
  pmodel = Seq2SeqModel(