davidberenstein1957's picture
feat: add stop_sequences to magpie generation
40e000b
raw
history blame
330 Bytes
import gradio as gr
from src.distilabel_dataset_generator.sft import demo
demo = gr.TabbedInterface(
[demo],
["Supervised Fine-Tuning"],
title="⚗️ Distilabel Dataset Generator",
head="⚗️ Distilabel Dataset Generator",
theme="ParityError/Interstellar",
)
if __name__ == "__main__":
demo.launch()