dvilasuero HF staff commited on
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
- btn_duplicate = get_duplicate_button()
300
-
301
- dataset_description = gr.Textbox(
302
  label="Provide a description of the dataset",
303
  value=DEFAULT_SYSTEM_PROMPT_DESCRIPTION,
304
  )
 
 
 
 
 
 
305
 
306
- btn_generate_system_prompt = gr.Button(
307
- value="🧪 Generate Sytem Prompt and Sample Dataset"
308
- )
309
-
310
- system_prompt = gr.Textbox(
311
- label="Provide or correct the system prompt",
312
  value=DEFAULT_SYSTEM_PROMPT,
313
  )
314
 
315
- btn_generate_sample_dataset = gr.Button(
316
- value="🧪 Generate Sample Dataset of 5 rows and a single turn",
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(