Update app.py

app.py

@@ -27,7 +27,7 @@ def translator_fn_baseline(text_in):
      input_ids = source_tokenizer.encode(text_in, return_tensors="pt")
      output_ids = model.generate(
          input_ids,
-         max_length=len(text_in.split())+1,
+         max_length=len(text_in.split())+3,
          bos_token_id=target_tokenizer.bos_token_id,
          eos_token_id=target_tokenizer.eos_token_id,
          pad_token_id=target_tokenizer.pad_token_id,
@@ -43,7 +43,7 @@ def translator_fn_roberta(text_in):
     input_ids_pretrained_roberta = source_tokenizer_pretrained_roberta.encode(text_in, return_tensors="pt")
     output_ids_pretrained_roberta = model_pretrained_roberta.generate(
         input_ids_pretrained_roberta,
-        max_length=len(text_in.split())+1,
+        max_length=len(text_in.split())+3,
         bos_token_id=target_tokenizer_pretrained_roberta.bos_token_id,
         eos_token_id=target_tokenizer_pretrained_roberta.eos_token_id,
         pad_token_id=target_tokenizer_pretrained_roberta.pad_token_id,