From 1639332f0603681c4db295451dc47ac904683e60 Mon Sep 17 00:00:00 2001 From: Hironsan Date: Tue, 18 May 2021 14:36:18 +0900 Subject: [PATCH] Update upload factory for speech to text --- backend/api/views/upload/factory.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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}')