update prompt prefix and suffix
Browse files- handler.py +4 -3
handler.py
CHANGED
|
@@ -8,10 +8,11 @@ import torch
|
|
| 8 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 9 |
|
| 10 |
PROMPT_PREFIX = (
|
| 11 |
-
"
|
| 12 |
-
"со
|
|
|
|
| 13 |
)
|
| 14 |
-
PROMPT_SUFFIX = "\n\nСовременный вариант:"
|
| 15 |
|
| 16 |
|
| 17 |
def _as_list(x: Union[str, List[str]]) -> List[str]:
|
|
|
|
| 8 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 9 |
|
| 10 |
PROMPT_PREFIX = (
|
| 11 |
+
"Ты – модель, которая строго переписывает дореформенный русский текст "
|
| 12 |
+
"в современную орфографию, не меняя смысл и пунктуацию. "
|
| 13 |
+
"Не добавляй комментарии и не переводь текст.\n\nТекст:\n"
|
| 14 |
)
|
| 15 |
+
PROMPT_SUFFIX = "\n\nСовременный орфографический вариант:"
|
| 16 |
|
| 17 |
|
| 18 |
def _as_list(x: Union[str, List[str]]) -> List[str]:
|