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(
|