Update README.md

README.md

@@ -45,7 +45,7 @@ text = "初めまして。よろしくお願いします。"
 def translate(text_src):
     embeddings = src_tokenizer(text_src, return_attention_mask=False, return_token_type_ids=False, return_tensors='pt')
     embeddings = {k: v for k, v in embeddings.items()}
-    output = model.generate(**embeddings)[0, 1:-1]
+    output = model.generate(**embeddings, max_length=500)[0, 1:-1]
     text_trg = trg_tokenizer.decode(output.cpu())
     return text_trg