Spaces:

NITHUB-AI
/

EHN-EnglisH-to-Nigerian-Pidgin-Translator

Running

Enutrof commited on Sep 5, 2023

Commit

701fdff

1 Parent(s): 24e1fce

Updated app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ def load_translator(model_name='Enutrof/marian-mt-en-pcm'):
     pmodel_args = Seq2SeqArgs()
     pmodel_args.max_length = 1024
     pmodel_args.length_penalty = 1
-    pmodel_args.num_beams = 20
     pmodel_args.num_return_sequences = 3
     pmodel = Seq2SeqModel(

     pmodel_args = Seq2SeqArgs()
     pmodel_args.max_length = 1024
     pmodel_args.length_penalty = 1
+    pmodel_args.num_beams = 50
     pmodel_args.num_return_sequences = 3
     pmodel = Seq2SeqModel(