Update main.py

main.py

@@ -17,16 +17,18 @@ 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. Ensure the entire text is rewritten:
+    Example input: "Summer is hot."
+    Example output: "Nothing beats lazy afternoons by the pool, sipping lemonade and watching the clouds drift by..."
+    
+    Rewrite the following text to sound like a warm, engaging blog post written by a passionate human. 
+    Use vivid imagery, personal anecdotes, and conversational language. Ensure the entire text is rewritten:
     {content}
     Rewrite:
     """
     inputs = tokenizer(prompt, return_tensors="pt", padding=True, truncation=True)
-    print("inputs:", inputs)  # 打印模型输入
-    output = model.generate(inputs["input_ids"], max_length=1024, num_beams=1, do_sample=True, temperature=0.7, top_k=50, early_stopping=True)
-    print("output:", output)  # 打印模型输出1
+    output = model.generate(inputs["input_ids"], max_length=512, num_beams=8, do_sample=False, temperature=0.7, top_k=50, early_stopping=True)
     decoded_output = tokenizer.decode(output[0], skip_special_tokens=True)
-    print("Model output:", decoded_output)  # 打印模型输出2
+    print("Model output:", decoded_output)  # 打印模型输出
     return decoded_output
 
 # API端点，接收内容并返回“人性化”后的文本