Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 30 |
-
|
| 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."""
|