Spaces:
Running
Running
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 |
)
|