minor changes to layout
Browse files
src/distilabel_dataset_generator/sft.py
CHANGED
|
@@ -291,31 +291,32 @@ with gr.Blocks(
|
|
| 291 |
) as app:
|
| 292 |
gr.Markdown(
|
| 293 |
"""
|
| 294 |
-
### Generate a high quality SFT dataset in a breeze using [🐦⬛MagPie](https://arxiv.org/abs/2406.08464) and [🦙Llama 3.1 - 70B Instruct](https://huggingface.co/meta-llama/Meta-Llama-3.1-70B-Instruct).
|
| 295 |
|
| 296 |
-
More information on distilabel and techniques can be found in the "FAQ" tab. The code can be found in the [Spaces repository](https://huggingface.co/spaces/argilla/distilabel-dataset-generator/tree/main).
|
| 297 |
"""
|
| 298 |
)
|
| 299 |
-
|
| 300 |
-
|
| 301 |
-
dataset_description = gr.Textbox(
|
| 302 |
label="Provide a description of the dataset",
|
| 303 |
value=DEFAULT_SYSTEM_PROMPT_DESCRIPTION,
|
| 304 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 305 |
|
| 306 |
-
|
| 307 |
-
|
| 308 |
-
)
|
| 309 |
-
|
| 310 |
-
system_prompt = gr.Textbox(
|
| 311 |
-
label="Provide or correct the system prompt",
|
| 312 |
value=DEFAULT_SYSTEM_PROMPT,
|
| 313 |
)
|
| 314 |
|
| 315 |
-
|
| 316 |
-
|
| 317 |
-
|
| 318 |
-
|
|
|
|
|
|
|
|
|
|
| 319 |
table = gr.Dataframe(label="Generated Dataset", wrap=True, value=DEFAULT_DATASET)
|
| 320 |
|
| 321 |
btn_generate_system_prompt.click(
|
|
|
|
| 291 |
) as app:
|
| 292 |
gr.Markdown(
|
| 293 |
"""
|
|
|
|
| 294 |
|
|
|
|
| 295 |
"""
|
| 296 |
)
|
| 297 |
+
dataset_description = gr.TextArea(
|
|
|
|
|
|
|
| 298 |
label="Provide a description of the dataset",
|
| 299 |
value=DEFAULT_SYSTEM_PROMPT_DESCRIPTION,
|
| 300 |
)
|
| 301 |
+
with gr.Row():
|
| 302 |
+
gr.Column(scale=1)
|
| 303 |
+
btn_generate_system_prompt = gr.Button(
|
| 304 |
+
value="Generate sample dataset"
|
| 305 |
+
)
|
| 306 |
+
gr.Column(scale=1)
|
| 307 |
|
| 308 |
+
system_prompt = gr.TextArea(
|
| 309 |
+
label="If you want to improve the dataset, you can tune the system prompt and regenerate the sample",
|
|
|
|
|
|
|
|
|
|
|
|
|
| 310 |
value=DEFAULT_SYSTEM_PROMPT,
|
| 311 |
)
|
| 312 |
|
| 313 |
+
with gr.Row():
|
| 314 |
+
gr.Column(scale=1)
|
| 315 |
+
btn_generate_sample_dataset = gr.Button(
|
| 316 |
+
value="Regenerate sample dataset",
|
| 317 |
+
)
|
| 318 |
+
gr.Column(scale=1)
|
| 319 |
+
|
| 320 |
table = gr.Dataframe(label="Generated Dataset", wrap=True, value=DEFAULT_DATASET)
|
| 321 |
|
| 322 |
btn_generate_system_prompt.click(
|