From a3100ff23408733fcfdd55cfa558c7d9a82feb18 Mon Sep 17 00:00:00 2001 From: Hironsan Date: Tue, 18 May 2021 14:40:06 +0900 Subject: [PATCH] Update download factory for speech to text --- backend/api/views/download/factory.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/backend/api/views/download/factory.py b/backend/api/views/download/factory.py index c1c65666..e99b9c98 100644 --- a/backend/api/views/download/factory.py +++ b/backend/api/views/download/factory.py @@ -1,7 +1,7 @@ from typing import Type from ...models import (DOCUMENT_CLASSIFICATION, IMAGE_CLASSIFICATION, SEQ2SEQ, - SEQUENCE_LABELING) + SEQUENCE_LABELING, SPEECH2TEXT) from . import catalog, repositories, writer @@ -11,6 +11,7 @@ def create_repository(project) -> repositories.BaseRepository: SEQUENCE_LABELING: repositories.SequenceLabelingRepository, SEQ2SEQ: repositories.Seq2seqRepository, IMAGE_CLASSIFICATION: repositories.FileRepository, + SPEECH2TEXT: repositories.FileRepository, } if project.project_type not in mapping: ValueError(f'Invalid project type: {project.project_type}')