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