refactor: limit batch size MagpieGenerator

src/distilabel_dataset_generator/apps/sft.py

@@ -274,11 +274,10 @@ with gr.Blocks(
             value="⚗️ Generate Full Dataset", variant="primary"
         )
 
-        # Add this line here, before the button click event
         success_message = gr.Markdown(visible=False)
 
     def show_success_message(repo_id_value):
-        return gr.update(
+        return gr.Markdown(
             value=f"""
             <div style="padding: 1em; background-color: #e6f3e6; border-radius: 5px; margin-top: 1em;">
                 <h3 style="color: #2e7d32; margin: 0;">Dataset Published Successfully!</h3>

src/distilabel_dataset_generator/pipelines/sft.py

@@ -156,6 +156,7 @@ def get_pipeline(num_turns, num_rows, system_prompt):
                     ],
                 },
             ),
+            batch_size=1,
             n_turns=num_turns,
             num_rows=num_rows,
             system_prompt=system_prompt,