Upload app.py

app.py

@@ -278,12 +278,12 @@ def rewrite_document(prompt):
             result += f"📝 原文摘要：\n{parsed_text[:300]}...\n\n"
             result += f"⚠️ 中文改写功能需要更大的模型支持，当前GPT-2主要支持英文。"
         else:
-            input_text = parsed_text[:300]
+            input_text = parsed_text[:2000]
             prompt_text = f"Please rewrite the following text according to this instruction: {prompt}\n\nOriginal text:\n{input_text}\n\nRewritten text:"
             
             result = generator(
                 prompt_text,
-                max_new_tokens=150,
+                max_new_tokens=500,
                 num_return_sequences=1,
                 temperature=0.8,
                 do_sample=True,