diff --git a/backend/api/views/upload/factory.py b/backend/api/views/upload/factory.py index 52f6b739..9453943d 100644 --- a/backend/api/views/upload/factory.py +++ b/backend/api/views/upload/factory.py @@ -1,5 +1,5 @@ from ...models import (DOCUMENT_CLASSIFICATION, IMAGE_CLASSIFICATION, SEQ2SEQ, - SEQUENCE_LABELING) + SEQUENCE_LABELING, SPEECH2TEXT) from . import catalog, data, dataset, label @@ -22,6 +22,7 @@ def get_dataset_class(format: str): catalog.Excel.name: dataset.ExcelDataset, catalog.CoNLL.name: dataset.CoNLLDataset, catalog.ImageFile.name: dataset.FileBaseDataset, + catalog.AudioFile.name: dataset.FileBaseDataset, } if format not in mapping: ValueError(f'Invalid format: {format}') @@ -34,6 +35,7 @@ def get_label_class(project_type: str): SEQUENCE_LABELING: label.OffsetLabel, SEQ2SEQ: label.TextLabel, IMAGE_CLASSIFICATION: label.CategoryLabel, + SPEECH2TEXT: label.TextLabel, } if project_type not in mapping: ValueError(f'Invalid project type: {project_type}')