Add Summarization to Pipelines (#3128)

* passing * Undo stupid chg * docs * undo rename * delete-cruft * only import if you have torch * Dont rely on dict ordering * Fix dict ordering upstream * docstring link * docstring link * remove trailing comma for 3.5 compat * new name * delegate kwarging * Update kwargs
2020-03-17 18:04:21 -04:00
parent 2b60a26b46
commit 38a555a83c
4 changed files with 129 additions and 1 deletions
--- a/src/transformers/init.py
+++ b/src/transformers/init.py
@@ -113,6 +113,7 @@ from .pipelines import (
    Pipeline,
    PipelineDataFormat,
    QuestionAnsweringPipeline,
+    SummarizationPipeline,
    TextClassificationPipeline,
    TokenClassificationPipeline,
    pipeline,