Update app.py

app.py

@@ -37,15 +37,14 @@ class RequestBody(BaseModel):
 # -------------------------------
 SYSTEM_PROMPT = """You are an expert academic editor.
 
-Your task is to rewrite the user's text so that it:
-- Reads like natural, human-written English.
+Rewrite the user's text so that it:
+- Reads naturally and human-like.
 - Has clear structure and good flow.
-- Uses a formal, academic tone (IEEE-style if technical).
-- Fixes grammar, redundancy, and awkward phrasing.
-- Preserves all original meaning and technical content.
-- Does NOT add new facts, equations, or claims.
-
-Output only the rewritten text. Do not include explanations or notes.
+- Uses a formal academic tone.
+- Improves grammar, clarity, and coherence.
+- DOES NOT add explanations, apologies, meta comments, or reasoning steps.
+- DOES NOT mention rewriting, editing, IEEE, or instructions.
+- Outputs ONLY the rewritten paragraph and nothing else.
 """
 
 
@@ -65,11 +64,11 @@ async def humanize(body: RequestBody):
     try:
         output = llm(
             prompt,
-            max_tokens=256,      # lower = faster, enough for paragraph-length
-            temperature=0.55,    # lower temp => more stable / less random
+            max_tokens=160,
+            temperature=0.55,
             top_p=0.9,
             repeat_penalty=1.05,
-            stop=["</s>", "[USER]", "[SYSTEM]"],
+            stop=["</s>", "[USER]", "[SYSTEM]", "[ASSISTANT]", "[/ASSISTANT]"],
         )
         result = output["choices"][0]["text"].strip()
         return {"result": result}