davidberenstein1957 HF staff commited on
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}. Potential labels: {', '.join(labels)}",
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
  )