Commit
·
3646907
1
Parent(s):
10b52aa
update textcat prompt
Browse files
src/synthetic_dataset_generator/apps/textcat.py
CHANGED
|
@@ -105,7 +105,7 @@ def generate_dataset(
|
|
| 105 |
is_sample=is_sample,
|
| 106 |
)
|
| 107 |
labeller_generator = get_labeller_generator(
|
| 108 |
-
system_prompt=f"{system_prompt}.
|
| 109 |
labels=labels,
|
| 110 |
multi_label=multi_label,
|
| 111 |
)
|
|
|
|
| 105 |
is_sample=is_sample,
|
| 106 |
)
|
| 107 |
labeller_generator = get_labeller_generator(
|
| 108 |
+
system_prompt=f"{system_prompt}. Optional labels: {', '.join(labels)}. Only select actually relevant labels.",
|
| 109 |
labels=labels,
|
| 110 |
multi_label=multi_label,
|
| 111 |
)
|