Update app.py

app.py

@@ -57,7 +57,7 @@ def _translate(text: str, src_lang: str, tgt_lang: str):
     for paragraph in paragraphs:
         translated_sentences = []
         input_tokens = (
-            tokenizer("Translate to Chinese:\n\n" + paragraph, return_tensors="pt")
+            tokenizer("Translate to Chinese. Direct output translation result without any explaination::\n\n" + paragraph, return_tensors="pt")
             .input_ids[0]
             .cpu()
             .numpy()
@@ -65,7 +65,7 @@ def _translate(text: str, src_lang: str, tgt_lang: str):
         )
         translated_chunk = model.generate(
             input_ids=torch.tensor([input_tokens]).to(device),
-            max_length=len(input_tokens) + 10000,
+            max_length=len(input_tokens) + 1000,
             num_return_sequences=1,
         )
         print(translated_chunk)