요약 길이 조정하기
pipeline() 함수에는 min_new_tokens와 max_new_tokens라는 두 가지 중요한 매개변수가 있어요. 이는 결과 요약문의 길이를 짧게, 더 길게, 또는 특정 단어 수 범위로 조정할 때 유용합니다. 저장 공간이 제한되어 있거나(예: 작은 저장소), 가독성을 높이거나, 요약의 품질을 개선하려는 경우에 이렇게 설정할 수 있어요.
이 연습에서는 두 매개변수를 좁은 범위와 넓은 범위로 각각 설정해, 짧은 요약기와 긴 요약기를 실험해 보겠습니다.
transformers 라이브러리의 pipeline과 original_text는 이미 로드되어 있습니다.
이 연습은 강의의 일부입니다
Hugging Face 활용하기
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
# Generate a summary of original_text between 1 and 10 tokens
short_summarizer = pipeline(task="summarization", model="cnicu/t5-small-booksum", ____=1, ____=10)
short_summary_text = ____(original_text)
print(short_summary_text[0]["summary_text"])