Spaces:
Runtime error
Runtime error
Commit
Β·
f6a1e43
1
Parent(s):
3445828
feat: Add basic layout textcat
Browse files
app.py
CHANGED
|
@@ -2,6 +2,7 @@ import gradio as gr
|
|
| 2 |
|
| 3 |
from src.distilabel_dataset_generator.apps.faq import app as faq_app
|
| 4 |
from src.distilabel_dataset_generator.apps.sft import app as sft_app
|
|
|
|
| 5 |
|
| 6 |
theme = gr.themes.Monochrome(
|
| 7 |
spacing_size="md",
|
|
@@ -25,8 +26,8 @@ css = """
|
|
| 25 |
"""
|
| 26 |
|
| 27 |
demo = gr.TabbedInterface(
|
| 28 |
-
[sft_app, faq_app],
|
| 29 |
-
["Supervised Fine-Tuning", "FAQ"],
|
| 30 |
css=css,
|
| 31 |
title="""
|
| 32 |
<style>
|
|
|
|
| 2 |
|
| 3 |
from src.distilabel_dataset_generator.apps.faq import app as faq_app
|
| 4 |
from src.distilabel_dataset_generator.apps.sft import app as sft_app
|
| 5 |
+
from src.distilabel_dataset_generator.apps.textcat import app as textcat_app
|
| 6 |
|
| 7 |
theme = gr.themes.Monochrome(
|
| 8 |
spacing_size="md",
|
|
|
|
| 26 |
"""
|
| 27 |
|
| 28 |
demo = gr.TabbedInterface(
|
| 29 |
+
[sft_app, textcat_app, faq_app],
|
| 30 |
+
["Supervised Fine-Tuning", "Text Classification", "FAQ"],
|
| 31 |
css=css,
|
| 32 |
title="""
|
| 33 |
<style>
|
src/distilabel_dataset_generator/apps/sft.py
CHANGED
|
@@ -21,7 +21,7 @@ from src.distilabel_dataset_generator.pipelines.sft import (
|
|
| 21 |
get_response_generator,
|
| 22 |
)
|
| 23 |
from src.distilabel_dataset_generator.utils import (
|
| 24 |
-
|
| 25 |
get_org_dropdown,
|
| 26 |
swap_visibilty,
|
| 27 |
)
|
|
@@ -233,24 +233,7 @@ def upload_pipeline_code(
|
|
| 233 |
progress(1.0, desc="Pipeline code uploaded")
|
| 234 |
|
| 235 |
|
| 236 |
-
|
| 237 |
-
.main_ui_logged_out{opacity: 0.3; pointer-events: none}
|
| 238 |
-
"""
|
| 239 |
-
|
| 240 |
-
with gr.Blocks(
|
| 241 |
-
title="𧬠Synthetic Data Generator",
|
| 242 |
-
head="𧬠Synthetic Data Generator",
|
| 243 |
-
css=css,
|
| 244 |
-
) as app:
|
| 245 |
-
with gr.Row():
|
| 246 |
-
gr.Markdown(
|
| 247 |
-
"Want to run this locally or with other LLMs? Take a look at the FAQ tab. distilabel Synthetic Data Generator is free, we use the authentication token to push the dataset to the Hugging Face Hub and not for data generation."
|
| 248 |
-
)
|
| 249 |
-
with gr.Row():
|
| 250 |
-
gr.Column()
|
| 251 |
-
get_login_button()
|
| 252 |
-
gr.Column()
|
| 253 |
-
|
| 254 |
gr.Markdown("## Iterate on a sample dataset")
|
| 255 |
with gr.Column() as main_ui:
|
| 256 |
dataset_description = gr.TextArea(
|
|
|
|
| 21 |
get_response_generator,
|
| 22 |
)
|
| 23 |
from src.distilabel_dataset_generator.utils import (
|
| 24 |
+
get_base_app,
|
| 25 |
get_org_dropdown,
|
| 26 |
swap_visibilty,
|
| 27 |
)
|
|
|
|
| 233 |
progress(1.0, desc="Pipeline code uploaded")
|
| 234 |
|
| 235 |
|
| 236 |
+
with get_base_app() as app:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 237 |
gr.Markdown("## Iterate on a sample dataset")
|
| 238 |
with gr.Column() as main_ui:
|
| 239 |
dataset_description = gr.TextArea(
|
src/distilabel_dataset_generator/apps/textcat.py
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from src.distilabel_dataset_generator.utils import get_base_app
|
| 2 |
+
|
| 3 |
+
with get_base_app() as app:
|
| 4 |
+
pass
|
src/distilabel_dataset_generator/utils.py
CHANGED
|
@@ -81,3 +81,28 @@ def swap_visibilty(oauth_token: OAuthToken = None):
|
|
| 81 |
return gr.update(elem_classes=["main_ui_logged_in"])
|
| 82 |
else:
|
| 83 |
return gr.update(elem_classes=["main_ui_logged_out"])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 81 |
return gr.update(elem_classes=["main_ui_logged_in"])
|
| 82 |
else:
|
| 83 |
return gr.update(elem_classes=["main_ui_logged_out"])
|
| 84 |
+
|
| 85 |
+
|
| 86 |
+
def get_base_app():
|
| 87 |
+
with gr.Blocks(
|
| 88 |
+
title="𧬠Synthetic Data Generator",
|
| 89 |
+
head="𧬠Synthetic Data Generator",
|
| 90 |
+
css=_LOGGED_OUT_CSS,
|
| 91 |
+
) as app:
|
| 92 |
+
with gr.Row():
|
| 93 |
+
gr.Markdown(
|
| 94 |
+
"Want to run this locally or with other LLMs? Take a look at the FAQ tab. distilabel Synthetic Data Generator is free, we use the authentication token to push the dataset to the Hugging Face Hub and not for data generation."
|
| 95 |
+
)
|
| 96 |
+
with gr.Row():
|
| 97 |
+
gr.Column()
|
| 98 |
+
get_login_button()
|
| 99 |
+
gr.Column()
|
| 100 |
+
|
| 101 |
+
gr.Markdown("## Iterate on a sample dataset")
|
| 102 |
+
with gr.Column() as main_ui:
|
| 103 |
+
pass
|
| 104 |
+
|
| 105 |
+
return app
|
| 106 |
+
|
| 107 |
+
|
| 108 |
+
_LOGGED_OUT_CSS = ".main_ui_logged_out{opacity: 0.3; pointer-events: none}"
|