bart
pip install torch transformers 4. 코드 설명BART 모델 (facebook/bart-large-cnn ) 불러오기 - 이 모델은 주로 텍스트 요약에 사용되지만, 재작성에도 활용 가능
입력 문장 토큰화 (tokenizer ) - 입력 텍스트를
max_length=1024 로 설정해 길이 제한을 적용
모델을 이용한 기사 재작성 (generate ) num_beams=4 : 빔 서치를 사용해 더 나은 결과 생성max_length=512 : 너무 긴 문장을 방지
결과 출력 (tokenizer.decode ) skip_special_tokens=True 옵션을 사용해 불필요한 토큰 제
추가 개선 가능 사항- 단순 요약이 아니라 문체나 표현을 더 바꿔야 한다면,
facebook/bart-base 를 사용하고,- Fine-tuning을 진행하여 맞춤형 모델을 학습 가능
- Hugging Face
T5 , GPT-3.5 등과 함께 사용하면 더 창의적인 변환 가능
|