Browse Source

Update download factory for speech to text

pull/1395/head
Hironsan 3 years ago
parent
commit
a3100ff234
1 changed files with 2 additions and 1 deletions
  1. 3
      backend/api/views/download/factory.py

3
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}')

Loading…
Cancel
Save