Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import gradio as gr
 import spaces
+import torch
 
 # 支持的语言选项
 LANGUAGES = {
@@ -144,7 +145,8 @@ def translate_text(text, source_lang, target_lang):
         )
         translated_chunk = model.generate(
             input_ids=torch.tensor([input_tokens]).to(device),
-            max_length=len(input_tokens) + 2048,
+            max_length=512,
+            num_beams=4,
             num_return_sequences=1,
         )
         full_output = tokenizer.decode(translated_chunk[0], skip_special_tokens=True)