Update upload factory for speech to text

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