diff --git a/backend/api/views/download/catalog.py b/backend/api/views/download/catalog.py index 078eb993..81bf788c 100644 --- a/backend/api/views/download/catalog.py +++ b/backend/api/views/download/catalog.py @@ -5,7 +5,7 @@ from pydantic import BaseModel from typing_extensions import Literal from ...models import (DOCUMENT_CLASSIFICATION, IMAGE_CLASSIFICATION, SEQ2SEQ, - SEQUENCE_LABELING) + SEQUENCE_LABELING, SPEECH2TEXT) from . import examples @@ -86,3 +86,6 @@ Options.register(SEQ2SEQ, JSONL, OptionNone, examples.Text_JSONL) # Image Classification Options.register(IMAGE_CLASSIFICATION, JSONL, OptionNone, examples.CategoryImageClassification) + +# Speech to Text +Options.register(SPEECH2TEXT, JSONL, OptionNone, examples.Speech2Text) diff --git a/backend/api/views/download/examples.py b/backend/api/views/download/examples.py index 1f0ade34..35b6a4ad 100644 --- a/backend/api/views/download/examples.py +++ b/backend/api/views/download/examples.py @@ -62,3 +62,12 @@ CategoryImageClassification = """ } ] """ + +Speech2Text = """ +[ + { + "filename": "20210514.mp3", + "label": ["Lorem ipsum dolor sit amet"] + } +] +"""