Commit
·
422059f
1
Parent(s):
af420e8
update sft to chat
Browse files
src/synthetic_dataset_generator/app.py
CHANGED
@@ -2,7 +2,7 @@ from synthetic_dataset_generator._tabbedinterface import TabbedInterface
|
|
2 |
|
3 |
# from synthetic_dataset_generator.apps.eval import app as eval_app
|
4 |
from synthetic_dataset_generator.apps.about import app as about_app
|
5 |
-
from synthetic_dataset_generator.apps.
|
6 |
from synthetic_dataset_generator.apps.textcat import app as textcat_app
|
7 |
|
8 |
theme = "argilla/argilla-theme"
|
@@ -21,8 +21,8 @@ button[role="tab"][aria-selected="true"]:hover {border-color: var(--button-prima
|
|
21 |
image = """<br><img src="https://raw.githubusercontent.com/argilla-io/synthetic-data-generator/main/assets/logo.svg" alt="Synthetic Data Generator Logo" style="display: block; margin-left: auto; margin-right: auto; width: clamp(50%, 400px, 100%)"/>"""
|
22 |
|
23 |
demo = TabbedInterface(
|
24 |
-
[textcat_app,
|
25 |
-
["Text Classification", "
|
26 |
css=css,
|
27 |
title=image,
|
28 |
theme=theme,
|
|
|
2 |
|
3 |
# from synthetic_dataset_generator.apps.eval import app as eval_app
|
4 |
from synthetic_dataset_generator.apps.about import app as about_app
|
5 |
+
from synthetic_dataset_generator.apps.chat import app as chat_app
|
6 |
from synthetic_dataset_generator.apps.textcat import app as textcat_app
|
7 |
|
8 |
theme = "argilla/argilla-theme"
|
|
|
21 |
image = """<br><img src="https://raw.githubusercontent.com/argilla-io/synthetic-data-generator/main/assets/logo.svg" alt="Synthetic Data Generator Logo" style="display: block; margin-left: auto; margin-right: auto; width: clamp(50%, 400px, 100%)"/>"""
|
22 |
|
23 |
demo = TabbedInterface(
|
24 |
+
[textcat_app, chat_app, about_app],
|
25 |
+
["Text Classification", "Chat", "About"],
|
26 |
css=css,
|
27 |
title=image,
|
28 |
theme=theme,
|
src/synthetic_dataset_generator/apps/{sft.py → chat.py}
RENAMED
@@ -24,11 +24,7 @@ from synthetic_dataset_generator.constants import (
|
|
24 |
MODEL,
|
25 |
SFT_AVAILABLE,
|
26 |
)
|
27 |
-
from synthetic_dataset_generator.pipelines.
|
28 |
-
get_embeddings,
|
29 |
-
get_sentence_embedding_dimensions,
|
30 |
-
)
|
31 |
-
from synthetic_dataset_generator.pipelines.sft import (
|
32 |
DEFAULT_DATASET_DESCRIPTIONS,
|
33 |
generate_pipeline_code,
|
34 |
get_magpie_generator,
|
@@ -36,6 +32,10 @@ from synthetic_dataset_generator.pipelines.sft import (
|
|
36 |
get_prompt_rewriter,
|
37 |
get_response_generator,
|
38 |
)
|
|
|
|
|
|
|
|
|
39 |
from synthetic_dataset_generator.utils import (
|
40 |
get_argilla_client,
|
41 |
get_org_dropdown,
|
|
|
24 |
MODEL,
|
25 |
SFT_AVAILABLE,
|
26 |
)
|
27 |
+
from synthetic_dataset_generator.pipelines.chat import (
|
|
|
|
|
|
|
|
|
28 |
DEFAULT_DATASET_DESCRIPTIONS,
|
29 |
generate_pipeline_code,
|
30 |
get_magpie_generator,
|
|
|
32 |
get_prompt_rewriter,
|
33 |
get_response_generator,
|
34 |
)
|
35 |
+
from synthetic_dataset_generator.pipelines.embeddings import (
|
36 |
+
get_embeddings,
|
37 |
+
get_sentence_embedding_dimensions,
|
38 |
+
)
|
39 |
from synthetic_dataset_generator.utils import (
|
40 |
get_argilla_client,
|
41 |
get_org_dropdown,
|
src/synthetic_dataset_generator/pipelines/{sft.py → chat.py}
RENAMED
File without changes
|