Spaces:
Runtime error
Runtime error
Commit
Β·
d2df8be
1
Parent(s):
342e737
remove obsolete class
Browse files
src/synthetic_dataset_generator/pipelines/textcat.py
CHANGED
@@ -60,19 +60,6 @@ class TextClassificationTask(BaseModel):
|
|
60 |
)
|
61 |
|
62 |
|
63 |
-
class DatasetDescription(BaseModel):
|
64 |
-
description: str = Field(
|
65 |
-
...,
|
66 |
-
title="description",
|
67 |
-
description="The description of the dataset.",
|
68 |
-
)
|
69 |
-
labels: list[str] = Field(
|
70 |
-
...,
|
71 |
-
title="labels",
|
72 |
-
description="The possible labels for the classification task.",
|
73 |
-
)
|
74 |
-
|
75 |
-
|
76 |
def get_prompt_generator():
|
77 |
prompt_generator = TextGeneration(
|
78 |
llm=InferenceEndpointsLLM(
|
|
|
60 |
)
|
61 |
|
62 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
63 |
def get_prompt_generator():
|
64 |
prompt_generator = TextGeneration(
|
65 |
llm=InferenceEndpointsLLM(
|