Update main.py

main.py

@@ -18,15 +18,15 @@ class ContentRequest(BaseModel):
 def generate_humanized_content(content: str) -> str:
     prompt = f"""
     Example input: "Summer is hot."
-    Example output: "Nothing beats lazy afternoons by the pool, sipping lemonade and watching the clouds drift by..."
+    Example output: "Summer is a sizzling season that loves to give us a warm, steamy welcome..."
     
-    Rewrite the following text to sound like a warm, engaging blog post written by a passionate human. 
+    change 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)
-    output = model.generate(inputs["input_ids"], max_length=512, num_beams=8, do_sample=False, temperature=0.7, top_k=50, early_stopping=True)
+    output = model.generate(inputs["input_ids"], max_length=512, num_beams=4, do_sample=True, temperature=0.7, early_stopping=True)
     decoded_output = tokenizer.decode(output[0], skip_special_tokens=True)
     print("Model output:", decoded_output)  # 打印模型输出
     return decoded_output