트랜스포머 파이프라인 예시

!pip install transformers --quiet
!pip install torch --quiet

from transformers import pipeline

classifier = pipeline("zero-shot-classification")
classifier(
    "This is a course about the Transformers library",
    candidate_labels=["education", "politics", "business"],
)

{'sequence': 'This is a course about the Transformers library',
 'labels': ['education', 'business', 'politics'],
 'scores': [0.8445963859558105, 0.111976258456707, 0.043427448719739914]}

from transformers import pipeline

generator = pipeline("text-generation")
generator("In this course, we will teach you how to")

#help(generator)
#print(generator)

[{'generated_text': 'In this course, we will teach you how to understand and use '
                    'data flow and data interchange when handling user data. We '
                    'will be working with one or more of the most commonly used '
                    'data flows ? data flows of various types, as seen by the '
                    'HTTP'}]

from transformers import pipeline

generator = pipeline("text-generation")
generator("In this course, we will teach you how to", num_return_sequences=2, max_length=15)