diff --git a/auto_labeling_pipeline/menu.py b/auto_labeling_pipeline/menu.py index 221a4d4..ba28541 100644 --- a/auto_labeling_pipeline/menu.py +++ b/auto_labeling_pipeline/menu.py @@ -91,7 +91,7 @@ def register(cls, task: Type[t.Task], model: Type[mo.RequestModel], template: Ty mp.AmazonRekognitionLabelDetectionTemplate ) Options.register( - t.Seq2seq, + t.SpeechToText, mo.GCPSpeechToTextRequestModel, mp.GCPSpeechToTextTemplate )