nllb-translation-demo

Build error

armanokka commited on Nov 29, 2022

Commit

b3c6542

•

1 Parent(s): 8b1898a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -41,7 +41,7 @@ def translation(source, target, text):
     tokenizer = model_dict[model_name + '_tokenizer']
     translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=source, tgt_lang=target)
-    output = translator(text, max_length=400)
     end_time = time.time()

     tokenizer = model_dict[model_name + '_tokenizer']
     translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=source, tgt_lang=target)
+    output = translator(text, max_length=4098)
     end_time = time.time()