Update main.py

main.py

@@ -17,12 +17,12 @@ class ContentRequest(BaseModel):
 # Humanize的Prompt
 def generate_humanized_content(content: str) -> str:
     prompt = f"""
-    Rewrite the following text to make it sound more human, engaging, and relatable:
-    Text: {content}
+    Rewrite the following text to make it sound more human, engaging, and relatable. Ensure the entire text is rewritten:
+    {content}
     Rewrite:
     """
     inputs = tokenizer(prompt, return_tensors="pt", padding=True, truncation=True)
-    output = model.generate(inputs["input_ids"], max_length=512, num_beams=4, do_sample=True, temperature=0.7, top_k=50, early_stopping=True)
+    output = model.generate(inputs["input_ids"], max_length=1024, num_beams=1, do_sample=True, temperature=0.7, top_k=50, early_stopping=True)
     decoded_output = tokenizer.decode(output[0], skip_special_tokens=True)
     print("Model output:", decoded_output)  # 打印模型输出
     return decoded_output