Supported Tasks and Pipelines

The task parameter defines the type of pipeline that will be returned. Below is a list of currently supported tasks and their corresponding pipelines:

• audio-classification: Returns an AudioClassificationPipeline.
• automatic-speech-recognition: Returns an AutomaticSpeechRecognitionPipeline.
• depth-estimation: Returns a DepthEstimationPipeline.
• document-question-answering: Returns a DocumentQuestionAnsweringPipeline.
• feature-extraction: Returns a FeatureExtractionPipeline.
• fill-mask: Returns a FillMaskPipeline.
• image-classification: Returns an ImageClassificationPipeline.
• image-feature-extraction: Returns an ImageFeatureExtractionPipeline.
• image-segmentation: Returns an ImageSegmentationPipeline.
• image-text-to-text: Returns an ImageTextToTextPipeline.
• image-to-image: Returns an ImageToImagePipeline.
• image-to-text: Returns an ImageToTextPipeline.
• mask-generation: Returns a MaskGenerationPipeline.
• object-detection: Returns an ObjectDetectionPipeline.
• question-answering: Returns a QuestionAnsweringPipeline.
• summarization: Returns a SummarizationPipeline.
• table-question-answering: Returns a TableQuestionAnsweringPipeline.
• text2text-generation: Returns a Text2TextGenerationPipeline.
• text-classification (alias: sentiment-analysis): Returns a TextClassificationPipeline.
• text-generation: Returns a TextGenerationPipeline.
• text-to-audio (alias: text-to-speech): Returns a TextToAudioPipeline.
• token-classification (alias: ner): Returns a TokenClassificationPipeline.
• translation: Returns a TranslationPipeline.
• translation_xx_to_yy: Returns a TranslationPipeline.
• video-classification: Returns a VideoClassificationPipeline.
• visual-question-answering: Returns a VisualQuestionAnsweringPipeline.
• zero-shot-classification: Returns a ZeroShotClassificationPipeline.
• zero-shot-image-classification: Returns a ZeroShotImageClassificationPipeline.
• zero-shot-audio-classification: Returns a ZeroShotAudioClassificationPipeline.
• zero-shot-object-detection: Returns a ZeroShotObjectDetectionPipeline.