Commit
•
c7f7750
1
Parent(s):
7dfd764
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(
|