loocorez commited on
Commit
769bf82
·
verified ·
1 Parent(s): 6af58a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -25,10 +25,10 @@ MODEL_NAME = "loocorez/reverse-text-warmup"
25
 
26
  # ---- prompt helpers --------------------------------------------------------
27
  parser = vf.XMLParser(["think", "answer"]) # <think> … </think>\n<answer> … </answer>
28
- SYSTEM_MSG = (
29
- "You are a helpful assistant that reverses the user's text. "
30
- f"Respond in exactly this format: {parser.get_format_str()}"
31
- )
32
 
33
  def build_prompt(user_msg: str, tok) -> str:
34
  """Use the model’s native chat template so all special tokens are right."""
 
25
 
26
  # ---- prompt helpers --------------------------------------------------------
27
  parser = vf.XMLParser(["think", "answer"]) # <think> … </think>\n<answer> … </answer>
28
+ SYSTEM_MSG = f"""Reverse the given text.
29
+
30
+ Respond in the following format:
31
+ {parser.get_format_str()}"""
32
 
33
  def build_prompt(user_msg: str, tok) -> str:
34
  """Use the model’s native chat template so all special tokens are right."""