Update main.py

main.py

@@ -22,9 +22,11 @@ def generate_humanized_content(content: str) -> str:
     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
     decoded_output = tokenizer.decode(output[0], skip_special_tokens=True)
-    print("Model output:", decoded_output)  # 打印模型输出
+    print("Model output:", decoded_output)  # 打印模型输出2
     return decoded_output
 
 # API端点，接收内容并返回“人性化”后的文本