Update main.py

main.py

@@ -16,7 +16,7 @@ class ContentRequest(BaseModel):
 
 # Humanize的Prompt
 def generate_humanized_content(content: str) -> str:
-    prompt = f"""
+    instructions = """
     You are a skilled writer tasked with rewriting AI-generated text to make it sound more human, engaging, and relatable. Please follow these instructions:
 
     **Humanizing Instructions:**
@@ -41,13 +41,13 @@ def generate_humanized_content(content: str) -> str:
     - Show vulnerability: Share imperfections, doubts, or fears to make the text more relatable and authentic.
     - Use humor: Humor can help humanize the text and make it more enjoyable to read.
     - Edit and refine: Review the text multiple times to refine the language, tone, and flow.
-
-    Now, please humanize the following text:
-    {content}
     """
+    prompt = f"{instructions}\n\nNow, please humanize the following text:\n{content}"
     inputs = tokenizer(prompt, return_tensors="pt", padding=True, truncation=True)
     output = model.generate(inputs["input_ids"], max_length=512, num_beams=4, early_stopping=True)
-    return tokenizer.decode(output[0], skip_special_tokens=True)
+    decoded_output = tokenizer.decode(output[0], skip_special_tokens=True)
+    print("Model output:", decoded_output)  # 打印模型输出
+    return decoded_output
 
 # API端点，接收内容并返回“人性化”后的文本
 @app.post("/humanize/")